- 链接:
https://leetcode.cn/problems/minimum-moves-to-equal-array-elements-ii/ - 题目:
给你一个长度为 n 的整数数组 nums ,返回使所有数组元素相等需要的最少移动数。
在一步操作中,你可以使数组中的一个元素加 1 或者减 1 。
示例:
输入:nums = [1,2,3]
输出:2
解释:
只需要两步操作(每步操作指南使一个元素加 1 或减 1):
[1,2,3] => [2,2,3] => [2,2,2]
- 思路:
- 假设数组最后都等于a。首先,需要搞清楚a值如何确定。我一开始以为a是平均数,但是提交后不通过。根据不通过例子,发现a是数组元素排序后的中位数。(如果数组偶数个,取中位数的其中一个)
- 先对数组排序,取中位数。可以直接使用
sort
函数,也可以使用快排等排序算法。 - 再遍历,求出最少移动数。
- 代码:
class Solution {
public:
int minMoves2(vector<int>& nums) {
int sum=0,ans=0,i=0,j=nums.size();
sort(nums.begin(),nums.end());
i=(i+j)/2;
sum=nums[i];
for(i=0,j--;i<j;i++,j--){
ans+=sum-nums[i];
ans+=nums[j]-sum;
}
return ans;
}
};