1,题目要求
Given a non-empty integer array, find the minimum number of moves required to make all array elements equal, where a move is incrementing a selected element by 1 or decrementing a selected element by 1.
给定一个数组序列,将其中的数字进行移动,判断所需要移动的总的最小距离。
2,题目思路
对于这道题,首先要搞清楚题意。所要移动的目标数字并不是需要数组中需要存在的数。题目是要找到使得所有元素都相等的总的移动的最小次数,其中移动将所选元素递增1或将所选元素递减1。
因此,对于两个数字,使得二者移动总和最下的移动方法,就是移动到二者中间的某一个位置。也就是说,对于每一对的数字(左右两端),移动的最优距离都是二者距离之差。
因此,我们只需要对这些距离进行计算即可得到最后的结果。
题目还有另外一种解决办法。因为上面的这种方法需要整个数组是按照从小到大的顺序,因此,整个程序开始之前是需要进行一次排序的。这也是一个比较大的时间开销。
因此,还有一种类似的思路,就是在排序上下功夫。也就是利用STL之中的nth_element方法:
这个方法有三个参数,即nth_element(fast, nth, last)。这个方法是使得第n位上正好是第n大的数字,然后在它前面的数字都是比它小的数字;在它后面都是比它大的数字。但是这些数字并没有完全排序。
因此它只是一种部分排序的方法。第一个和第三个参数表示的是范围。
在这道题目中,即:
nth_element(nums.begin(), nums.begin()+nums.size()/2, nums.end());
这样就使得大小位于中间的数字正好位于它应该在的位置上。之后也是这种思路,让前后和后面的数字都编程它即可。这样移动的距离相应的也就是最小的了。
3,程序源码
class Solution {
public:
int minMoves2(vector<int>& nums) {
sort(nums.begin(), nums.end());
int i = 0, j = nums.size() - 1, count = 0;
while (i < j)
count += (nums[j--] - nums[i++]);
return count;
}
};