leetcode 462. Minimum Moves to Equal Array Elements II

题目概述

解题思路

我一开始的思路是先求数组均值，然后和每个元素相比得到步长。但是我发现这个方法存在隐患，如果每个元素的值都取到，并且总共有个元素，那么在计算均值的时候，我们如果先对所有元素求和，这个和就会达到，超出了int型表示的范围，需要用long long类型。

接下来我看了别人的思路：有个老哥说，其实我们应该先观察一下这个数组的取到的中间值的规律。

观察发现，当数组元素有奇数个时，中间值就是它的中位数；当数组元素有偶数个时，中间值就是两个“中位数”及其间任意一个值。因此，我们只需要找到数组的中位数，就能算出这道题了。

方法性能

如果采用排序的方法处理数组，时间复杂度就是O()；实际上，也可以用复杂度为O(n)的方法找到中位数。

示例代码

class Solution {
public:
    int minMoves2(vector<int>& nums) 
    {
        int len = nums.size(), res = 0;
        int mid_idx = len % 2 ? ((len - 1) / 2) : (len / 2); 
        sort(nums.begin(), nums.end());
        int mid_val = nums[mid_idx];
        
        for(int i = 0; i < len; ++i)
        {
            if(i < mid_idx)
                res += mid_val - nums[i];
            else
                res += nums[i] - mid_val;
        }
        
        return res;
    }
};

线性时间复杂度的方法（实际很慢）：

class Solution {
public:
    int getMid(vector<int>& A, int ranks)
    {
        if(A.size() == 1)
            return A[0];
        
        vector<int> L, R;
        int temp_val = A[0];
        for(int i = 1; i < A.size(); ++i)
        {
            if(A[i] < temp_val)
                L.push_back(A[i]);
            else
                R.push_back(A[i]);
        }
        
        if(L.size() + 1 == ranks)
            return A[0];
        else if(L.size() + 1 < ranks)
            return getMid(R, ranks - L.size() - 1);
        else
            return getMid(L, ranks);
    }
    
    int minMoves2(vector<int>& nums) 
    {
        int len = nums.size();
        int mid_idx = int((len + 1) / 2);
        int mid_val = getMid(nums, mid_idx);
        
        int res = 0;
        for(int i = 0; i < len; ++i)
        {
            res += abs(nums[i] - mid_val);
        }
        
        return res;
        
    }
};