题目链接:https://leetcode-cn.com/problems/minimum-increment-to-make-array-unique/
题目描述
给定整数数组 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
代码
// 时间复杂度O(nlogn) 空间复杂度O(1)
class Solution {
public:
int minIncrementForUnique(vector<int>& A) {
if(A.size()<2) return 0;
sort(A.begin(), A.end());
int last = A[0], ret = 0;
for(int i = 1; i < A.size(); ++i){
if(A[i] <= last){
ret += last - A[i] +1; // 将A[i]调整为last+1所需次数
last ++;
}
else
last = A[i];
}
return ret;
}
};