题目:
给定整数数组 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
解题思路【统计】:
1、首先统计出每个数出现的次数,然后从小到大遍历每个数 i:
2、如果 i 出现了两次以上,那么只留下1个,其他的count-1个数记录下来;
3、如果 i 没有出现过,那么在记录下来的数中选取一个 j,将它增加到 j,需要进行的操作次数为 i - j。
代码实现:
class Solution {
public int minIncrementForUnique(int[] A) {
int[] count = new int[80000];
for (int i : A) {
count[i]++;
}
int cnt = 0,move = 0;
for (int i = 0; i < count.length; i++) {
if(count[i] > 1){
int d = count[i] - 1;
move += d;
cnt -= i*d;
}else if(move > 0 && count[i] == 0){
move--;
cnt += i;
}
}
return cnt;
}
}
复杂度分析:
时间复杂度:O(N),其中N的数量级是数组 A 的长度加上其数据范围内的最大值,因为在最坏情况下,数组 A 中的所有数都是数据范围内的最大值。
空间复杂度:O(N),需要长度N的数组统计每个数出现的次数。
参考解题思路【线性探测】:
思路:hash表线性探测
如果地址冲突了,会探测它的下一个位置,如果下一个位置还是冲突,继续向后看,直到第一个不冲突的位置为止
探测优化思路:路径压缩
经过某条路径最终探测到一个空位置x后,将这条路径上的值都变成空位置所在的下标x, 假如下次探测的点又是这条路径上的点,
则可以直接跳转到这次探测到的空位置x,从x开始继续探测
代码实现:
class Solution {
int[] ans = new int [80000];
public int minIncrementForUnique(int[] A) {
Arrays.fill(ans, -1); // -1表示空位
int move = 0;
for (int a: A) {
int b = findPos(a);
move += b - a;//记录移动的步数
}
return move;
}
// 线性探测寻址(含路径压缩)
private int findPos(int a) {
int b = ans[a];
// 如果a对应的位置pos[a]是空位,直接放入
if (b == -1) {
ans[a] = a;
return a;
}
//如果当前非空位,递归寻找其他空位
b = findPos(b + 1);
ans[a] = b; // 路径压缩原理
return b;
}
}