题目:
给你一个长度为 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
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/minimum-moves-to-equal-array-elements-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
这道题就是现将数组进行排序,然后找到他们中间的那个元素的值,让其余每一个元素都变成这个中间元素的值,统计他的个数即可。
代码:
class Solution {
public:
int minMoves2(vector<int>& nums) {
sort(nums.begin(), nums.end());
int N = nums.size();
int mid = nums[N / 2];
int res = 0;
for (int n : nums) {
res += abs(n - mid);
}
return res;
}
};
注:
1.nth_element()函数
这个函数后面有三个参数,假设为nth_element(a,b,c)
这个意思是将数组从下标a-c这个范围内,将第b大的那个元素放到b的位置上,左边是比b小的元素,右边是比b大的元素。
如果要反过来的话,就在c后面的位置写为mycomp1。