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]
题目链接
题目原意是每次使n-1个元素加1,使得最后所有的数相等。容易想到,n-1个元素+1时,不会对最大数进行+1操作,否则就永远不可能到所有数相等的时候了。而对n-1个非最大数加1,可以等效为每次对最大数减1,直到所有数相等,而相等时容易知道,所有数均为数组最小数。故题目又可等效为,数组中所有元素与最小数之差的和,代码如下:
public class Solution {
public int minMoves(int[] nums) {
int min = Integer.MAX_VALUE;
for(int n : nums) min = Math.min(n, min);
int moves = 0;
for(int n : nums) moves += n - min;
return moves;
}
}
容易观察到,每次求各个数最小值之和,都要减掉最小数,所以可以将代码简化成数组所有数相加,减掉n个最小数,代码如下:
public class Solution {
public int minMoves(int[] nums) {
int min = Integer.MAX_VALUE;
for(int n : nums) min = Math.min(n, min);
int moves = 0;
for(int n : nums) moves += n;
return moves - nums.length * min;
}
}
在上面的代码中,找最小数并没有采用sort之后取第一个数的作法,这是因为其算法复杂度为 O(nlogn) ,而直接调用API的min来找,则复杂度仅为 O(n)