feat(interview_task): adds a solution to calculate the min num of ope…

interviewCodingAssesment.py

@@ -0,0 +1,116 @@
+"""
+You are given an array of integers nums and a non-negative integer k. 
+You can perform operations where you increase or decrease any element 
+of the array by 1. Your task is to determine the minimum number of operations 
+required to adjust the elements of nums such that the median of the array 
+becomes exactly k.
+
+The median is defined as the middle element in the sorted version of the array. 
+If the array has an even number of elements, the median is the larger of the 
+two middle values.
+
+Examples
+
+Input: nums = [3,7,1,4,6], k = 5
+Output: 1
+Explanation: After sorting, the array becomes [1,3,4,6,7]. 
+The median is the 3rd element (4). Increasing it to 5 requires 1 operation.
+
+Input: nums = [1,4,5,9,10], k = 8
+Output: 3
+Explanation: After sorting, the median is the 3rd element (5). 
+Increasing it to 8 requires 3 operations.
+
+Input: nums = [10,12,14,16,18], k = 14
+Output: 0
+Explanation: The median is already 14, so no operations are needed.
+
+Constraints
+
+1 <= nums.length <= 2 * 10^5
+"""
+from typing import List
+from statistics import median
+class Solution:
+    def minOperations(self, nums: List[int], k: int) -> int:
+        """
+        Calculate minimum operations to make k the median of nums.
+
+        Args:
+            nums: List of integers
+            k: Target median value
+
+        Returns:
+            int: Minimum number of operations required
+        """
+        # Your implementation here
+        nums.sort()
+        res = 0
+        my_median = median(nums)
+        if(my_median == k):
+            res = 0
+        elif my_median < k:
+            res = k - my_median
+        else:
+            res = my_median -k
+        return res
+
+def test_cases():
+    solution = Solution()
+    tests_passed = 0
+    total_tests = 3
+
+    # Test case 1
+    nums1 = [3, 7, 1, 4, 6]
+    k1 = 5
+    expected1 = 1
+    result1 = solution.minOperations(nums1, k1)
+    print(f"Test case 1: nums = {nums1}, k = {k1}")
+    print(f"Your output: {result1}")
+    print(f"Expected output: {expected1}")
+    if result1 == expected1:
+        print("✓ Test case 1 passed")
+        tests_passed += 1
+    else:
+        print("✗ Test case 1 failed")
+    print()
+
+    # Test case 2
+    nums2 = [1, 4, 5, 9, 10]
+    k2 = 8
+    expected2 = 3
+    result2 = solution.minOperations(nums2, k2)
+    print(f"Test case 2: nums = {nums2}, k = {k2}")
+    print(f"Your output: {result2}")
+    print(f"Expected output: {expected2}")
+    if result2 == expected2:
+        print("✓ Test case 2 passed")
+        tests_passed += 1
+    else:
+        print("✗ Test case 2 failed")
+    print()
+
+    # Test case 3
+    nums3 = [10, 12, 14, 16, 18]
+    k3 = 14
+    expected3 = 0
+    result3 = solution.minOperations(nums3, k3)
+    print(f"Test case 3: nums = {nums3}, k = {k3}")
+    print(f"Your output: {result3}")
+    print(f"Expected output: {expected3}")
+    if result3 == expected3:
+        print("✓ Test case 3 passed")
+        tests_passed += 1
+    else:
+        print("✗ Test case 3 failed")
+    print()
+
+    # Print summary
+    print(f"Test Results: {tests_passed} out of {total_tests} tests passed")
+    if tests_passed == total_tests:
+        print("Congratulations! All tests passed! 🎉")
+    else:
+        print(f"Keep trying! {total_tests - tests_passed} test(s) failed.")
+
+if __name__ == "__main__":
+    test_cases()