leetcode462.最少移动次数使数组元素相等2

题目大意

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

例如:

输入:
[1,2,3]
输出:
2
说明：
只有两个动作是必要的（记得每一步仅可使其中一个元素加1或减1）： 

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

解题思路

大致思路就是：首先排序，然后计算每个位置，判断是否是最优点。当前位置向右移动时，将右边的数字的结果减掉一个值，左边的数字的结果加上一个值。

class Solution {
public:
    int minMoves2(vector<int>& nums) {
    	if (nums.size() < 2)
    		return 0;

    	sort(nums.begin(), nums.end());

    	long length = nums.size(), res = LONG_MAX;
    	long curNum = 0;
    	for (int i = 1; i < length; ++i)
    		curNum += long(nums[i] - nums[0]);

    	for (int i = 1; i < length; ++i){
    		curNum = curNum - (length - i) * (nums[i] - nums[i - 1]) + i * (nums[i] - nums[i - 1]);
    		res = min(res, curNum);
    	}
    	return res;
    }
};