版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/sd4567855/article/details/86648851
day27, 使数组唯一的最小增量
题目来源:leetcode
给定整数数组 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 操作是不能让数组的每个值唯一的。
解答:该问题的关键在于如何达到 最小操作次数。
首先排序,之后逐个遍历,对相邻的两个数字进行判断。
代码:
int minIncrementForUnique(vector<int>& A) {
sort( A.begin(), A.end());
set<int, greater<int> > record;
int result = 0;
for( int i = 0; i < A.size(); i++)
if( record.find( A[i]) == record.end())
record.insert( A[i]);
else{
result += *record.begin() - A[i] + 1;
A[i] = *record.begin() + 1;
record.insert( A[i]);
}
return result;
}