原題
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.
You may assume the array's length is at most 10,000.
Example:
Input: [1,2,3] Output: 2 Explanation: Only two moves are needed (remember each move increments or decrements one element): [1,2,3] => [2,2,3] => [2,2,2]
解題思路
一开始以为是要求平均值,尝试用几个案例验证一下却不太对。之后又想了一下选择重复最多的一个数字,也不对。然后发现用中位数的结果是对的,就先把数组排序一下,然后算出每个数字与中位数的方差就好了。
代碼
#include<math.h>
#include <algorithm>
class Solution {
public:
int minMoves2(vector<int>& nums) {
std::sort(nums.begin(), nums.begin()+nums.size());
int middle, i, sum = 0;
if (nums.size() % 2 == 0) {
middle = nums.size()/2 - 1;
}
else {
middle = nums.size()/2;
}
for (i = 0; i < nums.size(); i++) {
sum += abs(nums[middle] - nums[i]);
}
return sum;
}
};
感想
做了几星期的题,感觉我做的的题都不是太难,太多是要运用你的数学能力而不是编程技巧。只要你有正确的思路,代码实现起来都很简单。