原题目链接:LeetCode945
分类
LeetCode
题意
给定整数数组 A,每次 move 操作将会选择任意 A[i]
,并将其递增 1
。
返回使 A
中的每个值都是唯一的最少操作次数。
样例输入输出
输入:[1,2,2]
**输出:**1
**解释:**经过一次 move 操作,数组将变为 [1, 2, 3]。
输入:[3,2,1,2,1,7]
**输出:**6
**解释:**经过 6 次 move 操作,数组将变为 [3, 4, 1, 2, 5, 7]。
可以看出 5 次或 5 次以下的 move 操作是不能让数组的每个值唯一的。
想法
看代码,比较容易理解
代码
/**
* Author: GatesMa
* Email: [email protected]
* Todo: ACM Training
* Date:2018/12/2
*/
class Solution {
public:
int minIncrementForUnique(vector<int>& A) {
int sum = 0;
if(A.size() == 0) return 0;
sort(A.begin(), A.end());
int k = A[0];
for(int i =1;i < A.size();i++){
k++;
if (k <= A[i]) {
k = A[i];
} else {
sum += k - A[i];
}
}
return sum;
}
};