3107. Minimum Operations to Make Median of Array Equal to K

You are given an integer array nums and a non-negative integer k. In one operation, you can increase or decrease any element by 1.

Return the minimum number of operations needed to make the median of nums equal to k.

The median of an array is defined as the middle element of the array when it is sorted in non-decreasing order. If there are two choices for a median, the larger of the two values is taken.

class Solution {
public:
    long long minOperationsToMakeMedianK(vector<int>& A, int k) {
        sort(begin(A), end(A));
        int n = A.size();
        int l2 = A.size() / 2;
        long long res = 0;
        for(int i = 0; i < l2; i++) {
            if(A[i] > k) {
                res += A[i] - k;
                A[i] = k;
            }
        }
        for(int i = l2 + 1; i < A.size(); i++) {
            if(A[i] < k) {
                res += k - A[i];
                A[i] = k;
            }
        }

        return res + abs(k - A[l2]);
    }
};