给定整数数组 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
func minIncrementForUnique(A []int) int {
cnt := 0
if len(A) == 0 {
return cnt
}
size := 400001
cnts := make([]int, size)
for i := range A {
cnts[A[i]]++
}
move := 0
for i := range cnts {
if cnts[i] > 1 {
// 要找的空位数
loops := cnts[i] - 1
for j := 0; j < loops; j++{
flag := false
for k := i + 1; k < size; k++{
move++
if cnts[k] == 0 {
cnts[k]++
flag = true
break
}
}
// 空位不够
if !flag {
return -1
}
}
}
}
return move
}