给定整数数组 A,每次 move 操作将会选择任意 A[i],并将其递增 1。
返回使 A 中的每个值都是唯一的最少操作次数。
示例 1:
输入:[1,2,2]
输出:1
解释:经过一次 move 操作,数组将变为 [1, 2, 3]。
示例 2:
输入:[3,2,1,2,1,7]
输出:6
解释:经过 6 次 move 操作,数组将变为 [3, 4, 1, 2, 5, 7]。
可以看出 5 次或 5 次以下的 move 操作是不能让数组的每个值唯一的。
提示:
0 <= A.length <= 40000
0 <= A[i] < 40000
思路:
数组长度不超过40000,最差情况下数组里的值全为40000,因此构建一个长度为80000的数组。对A进行计数,超过一次的把超出的次数给下一个,计数次数。
class Solution {
public int minIncrementForUnique(int[] A) {
if (A == null || A.length == 0) return 0;
int[] B = new int[80000];
for (int i = 0; i < A.length; i++)
B[A[i]]++;
int number = 0;
for (int i = 0; i < B.length-1; i++) {
int count = B[i];
if (count > 1) {
B[i+1] += count - 1;
number += count - 1;
}
}
return number;
}
}