最少移动次数使数组元素相等

题目描述

给定一个长度为 n 的非空整数数组，找到让数组所有元素相等的最小移动次数。每次移动可以使 n - 1 个元素增加 1。

示例

输入:
[1,2,3]

输出:
3

解释:
只需要3次移动（注意每次移动会增加两个元素的值）：

[1,2,3]  =>  [2,3,3]  =>  [3,4,3]  =>  [4,4,4]

思路

如果要达到全数组相等，那么n-1个+1==>1个-1； 所以为了把各个数字调整到相对值相等，要把每个数组减到最小的那个。 综合来看是：数组求和-最小值*数组长。

int minMoves(vector<int>& nums) {

	int result = 0;
	int minNum = INT_MAX;
	for (int i = 0; i < nums.size(); i++)
	{
		minNum = min(minNum, nums[i]);
	}

	for (int i = 0; i < nums.size(); i++)
	{
		result += nums[i] - minNum;
	}

	return result;
}

 

题目描述

给定一个非空整数数组，找到使所有数组元素相等所需的最小移动数，其中每次移动可将选定的一个元素加1或减1。 您可以假设数组的长度最多为10000。

示例

输入:
[1,2,3]

输出:
2

说明：
只有两个动作是必要的（记得每一步仅可使其中一个元素加1或减1）： 

[1,2,3]  =>  [2,2,3]  =>  [2,2,2]

思路

中位数定理，所有的数移到中位数代价最小。

证明

我们先假设一共有2n+1个数，它们从小到大排序之后如下：

 . . . a m b . . .

其中m是中位数。此时，m左边有n个数，m右边也有n个数。我们假设把m左边所有数变成m需要的代价是x，把m右边所有数变成m的代价是y，此时的总代价就是t = x+y

好，如果你觉得中位数不是最优解，我们来看看把所有数都变成a的总代价是多少。 由于之前m右边n个数变成m的代价是y，现在让右边的数全变成a，此时右边的数的代价是y+(m-a)*n；m左边的n个数全变成a，它们的代价会减少到x-(m-a)*n。所以两边相加，结果还是 x-(m-a)*n + y+(m-a)*n == x+y。 但是，别忘了，m也要变成a，所以总代价是x+y+m-a，大于x+y。同理，如果让所有数都变成比m大的b，总代价则变为x+y+b-m（你可以自己算一下），依然比x+y大。并且越往左移或者往右移，这个值都会越来越大。 因此，在有2n+1个数的时候，选择中位数就是最优解。

这个证明同样可以很简单地推广到2n个数。

. . . a b . . .

假设a左边有n-1个数，b右边也有n-1个数。如果我们选择把所有数都变成a，设a左边所有数变成a的代价是x，b右边所有数变成a的代价是y，因此总代价是x+y+b-a（b也要变成a）。 现在尝试下如果把所有数变成b，那么a左边的总代价变成了x+(b-a)*(n-1)，b右边总代价变成了y-(b-a)*(n-1)，同时还要把a变成b，因此总代价同样为x+(b-a)*(n-1)+y-(b-a)*(n-1) == x+y+b-a。也就是说当总个数为2n时，两个中位数选哪个最终结果都一样，但是继续左移和继续右移，都会使总代价增加（可以自己试试）。

至此，证明了中位数是最优策略

因此，这道题你只需要找到中位数，然后简单计算一下就能得出结果了

int minMoves2(vector<int>& nums) {

	int res = 0;
    sort(nums.begin(), nums.end());
	int i = 0;
	int j = nums.size() - 1;
	while (i < j)
	{
		res += nums[j--] - nums[i++];
	}

	return res;
}