945. 使數組唯一的最小增量 題目鏈接
解法一:
對數組 A 升序排序,count 記錄需要操作的次數。
若 A[i]<=A[i-1],count+=A[i-1]+1-A[i],A[i]=A[i-1]+1。
public int minIncrementForUnique(int[] A) {
Arrays.sort(A);
int count = 0;
for (int i = 1; i < A.length; i++) {
if (A[i] <= A[i - 1]) {
count += A[i - 1] + 1 - A[i];
A[i] = A[i - 1] + 1;
}
}
return count;
}
解法二:
建立哈希表 set,存儲數組無重複元素,count 記錄操作次數 (使用 boolean[50000] 代替)
依序遍歷數組 A,若元素在 set 中不存在則遍歷下一元素
若已存在,則 元素值、count++,直到在 set中無重複再遍歷下一元素
public int minIncrementForUnique(int[] A) {
boolean[] used = new boolean[50000];
int count = 0, index = 0;
while (index < A.length) {
if (!used[A[index]]) {
used[A[index]] = true;
index++;
} else {
A[index]++;
count++;
}
}
return count;
}