Given a non-empty integer array of size n, find the minimum number of moves required to make all array elements equal, where a move is incrementing n - 1 elements by 1.
Example:
Input:
[1,2,3]
Output:
3
Explanation:
Only three moves are needed (remember each move increments two elements):
[1,2,3] => [2,3,3] => [3,4,3] => [4,4,4]
解答:
设经过m步,最终每一个数都达到了x。其中n为nums的个数,sum为加之前的和,minnum为加之前的最小数。那么
sum+m(n-1)=x·n
x=minnum+m
于是
m=sum-min·n
代码:
class Solution {
public:
int minMoves(vector<int>& nums) {
int sum=0;
int minnum=nums[0];
for(int i=0;i<nums.size();i++)
{
sum+=nums[i];
minnum=min(minnum,nums[i]);
}
return sum-minnum*nums.size();
}
};
二刷时,发现有更优解,就是反过来理解,把n-1个数加相当于把另外一个数减1。
正确的解法相当的巧妙,需要换一个角度来看问题,其实给n-1个数字加1,效果等同于给那个未被选中的数字减1,比如数组[1,2,3], 给除去最大值的其他数字加1,变为[2,3,3],我们全体减1,并不影响数字间相对差异,变为[1,2,2],这个结果其实就是原始数组的最大值3自减1,那么问题也可能转化为,将所有数字都减小到最小值,这样难度就大大降低了,我们只要先找到最小值,然后累加每个数跟最小值之间的差值即可,参见代码如下
class Solution {
public:
int minMoves(vector<int>& nums) {
int result=0;
int min_num=INT_MAX;
for(auto num:nums){
min_num=min(num,min_num);
}
for(auto num:nums){
result+=num-min_num;
}
return result;
}
};