问题:
难度:medium
说明:
给出一个数组,然后只允许数组元素 + 1 或者 - 1,每一次都算一个操作,那么求出让数组所有元素数值都一样的时候,需要多少次操作。
题目连接:https://leetcode.com/problems/minimum-moves-to-equal-array-elements-ii/
输入范围:
n == nums.length
1 <= nums.length <= 105
-109 <= nums[i] <= 109
输入案例:
Example 1:
Input: nums = [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]
Example 2:
Input: nums = [1,10,2,9]
Output: 16
我的代码:
假设数组是有顺序的, 1 2 3 4 5 6 7 8 假设所有数都在一条直线上,那么画图就会存在:
那么 1 8 这两个头尾的数字,就必须移动到 n ,其中 1 <= n <= 8,才能让两个数字是相等并且移动次数最小的,因为 1 8 移动的时候:
两个数字无论移动到他们之间那个地方,他们的路程总和不变都是 8 - 1 = 7,也就是操作次数都是 7,这就可以认为是物理学中的路程不变,那么 1 8、2 7、3 6、4 5 这数据两两匹配,他们都必须往两者之间的点进行移动才保证操作次数最小,也就是路程最小,所以选择移动到 4 或者 5 就能保证最小操作数:
所以直接排序,然后统计下路程就能算出最小操作数:
Java:
class Solution {
public int minMoves2(int[] nums) {
Arrays.sort(nums);
int mid = nums[nums.length >> 1];
int res = 0;
for(int i : nums) res += Math.abs(i - mid);
return res;
}
}
C++:
class Solution {
public:
int minMoves2(vector<int>& nums) {
sort(nums.begin(), nums.end());
int mid = nums[nums.size() >> 1], res = 0;
for(int i = 0, len = nums.size(); i < len; i ++) res += abs(nums[i] - mid);
return res;
}
};