462. 最少移动次数使数组元素相等 II
给你一个长度为 n 的整数数组 nums ,返回使所有数组元素相等需要的最少移动数。
在一步操作中,你可以使数组中的一个元素加 1 或者减 1 。
示例 1:
输入:nums = [1,2,3]
输出:2
解释:
只需要两步操作(每步操作指南使一个元素加 1 或减 1):
[1,2,3] => [2,2,3] => [2,2,2]
示例 2:
输入:nums = [1,10,2,9]
输出:16
提示:
n == nums.length
1 <= nums.length <= 105
-109 <= nums[i] <= 109
思路:上来先想到的是变换的目标值是在数组的数字范围内,于是想着暴力了一波,用两个循环不断比较变换成当前数字的步数,得出最小值,其中步数是求出其他数字与当前数字的距离,然后加和。结果还是无法过全部样例
看了其他人的解答,大概是分两种情况考虑,目标值在数组范围之外和在数组范围之内,可以通过简单例子就能排除在范围之外的情况,而在范围之内的时候,当把数组排好序后,变化到最小值和最大值的步数是一致的,同时最小值和最大值有相当于排除它们之后的子数组的一个范围之外的目标,所以也是不可取的,通过不断缩小范围,发现满足最小的移动数的只能是中位数,所以题目变成数组中每个数字变化到中位数需要移动的步数,求中位数的话,如果不追求O(n),可以直接对数组排序,取nums[nums.length/2] 。具体过程可以查看:负雪明烛的解答
暴力代码,只过了18个样例
class Solution {
public int minMoves2(int[] nums) {
int min = Integer.MAX_VALUE;
for(int i=0;i<nums.length;i++){
int sum = 0;
for(int j=0;j<nums.length;j++){
sum+=Math.abs(nums[j]-nums[i]);
}
min = Math.min(sum,min);
}
return min;
}
}
正确解法
class Solution {
public int minMoves2(int[] nums) {
Arrays.sort(nums);
int res = 0;
for(int i=0;i<nums.length;i++){
res+=Math.abs(nums[i]-nums[nums.length/2]);
}
return res;
}
}