LeetCode 462. 最少移动次数使数组元素相等 II--前缀和

462. 最少移动次数使数组元素相等 II

给定一个非空整数数组，找到使所有数组元素相等所需的最小移动数，其中每次移动可将选定的一个元素加1或减1。 您可以假设数组的长度最多为10000。

例如:

输入:
[1,2,3]

输出:
2

说明：
只有两个动作是必要的（记得每一步仅可使其中一个元素加1或减1）：

[1,2,3] => [2,2,3] => [2,2,2]

题解

排序后计算前缀和，贪心策略，我们从左到右遍历，假设把所有元素变成nums[i]，那么左边的元素肯定都是加，右边的元素都是减，利用前缀和可以很快计算得到。

AC代码

class Solution {
public:
    typedef long long ll;
    ll sum[10010];
    int minMoves2(vector<int>& nums) 
    {
        sort(nums.begin(),nums.end());
        for(int i=0;i<nums.size();i++)
        sum[i+1]=sum[i]+nums[i];
        ll res=1e18;
        for(int i=1;i<=nums.size();i++)
        {
            ll left=nums[i-1]*(ll)(i-1)-sum[i-1];
            ll right=sum[nums.size()]-sum[i]-nums[i-1]*(ll)(nums.size()-i);
            res=min(res,left+right);
        }
        return int(res);
    }
};