LeetCode 462: Minimum Moves to Equal Array Elements II

LeetCode 462: Minimum Moves to Equal Array Elements II

题目描述

给定一个非空的整数数组，找到使所有数组元素相等所需的最小移动数，其中移动是将选定元素增加1或将选定元素减少1。

例：
输入：[1,2,3]
输出：2

说明：
仅需要两个动作（记住每个动作递增或递减一个元素）：[1,2,3] => [2,2,3] => [2,2,2]

解题思路

这是个典型的相遇问题，移动距离最小的方式是所有元素都移动到中位数。理由如下：

设 m 为中位数。a 和 b 是 m 两边的两个元素，且 b > a。要使 a 和 b 相等，它们总共移动的次数为 b - a，这个值等于 (b - m) + (m - a)，也就是把这两个数移动到中位数的移动次数。设数组长度为 N，则可以找到 N/2 对 a 和 b 的组合，使它们都移动到 m 的位置。

AC代码

class Solution {
public:
	int minMoves2(vector<int>& nums) {
		int n = nums.size();
		if (n <= 1)
			return 0;
		sort(nums.begin(), nums.end());
		int i = 0, j = n - 1;
		int move = 0;
		while (i < j)
		{
			move += nums[j] - nums[i];
			++i;
			--j;
		}
		return move;
	}
};