Leetcode-945 Minimum Increment to Make Array Unique(使数组唯一的最小增量)
给定整数数组 A,每次 move 操作将会选择任意 A[i]
,并将其递增 1
。
返回使 A
中的每个值都是唯一的最少操作次数。
示例 1:
输入:[1,2,2]
输出:1
解释:经过一次 move 操作,数组将变为 [1, 2, 3]。
1. 完全按照常规思路,超时!但是代码写完无错误!56 / 59 个通过测试用例
class Solution {
public:
int minIncrementForUnique(vector<int>& A) {
int N=A.size();
stack<int> si;
set<int> ss;
int count=0;
for(int i=0;i<N;i++)
{
if(ss.count(A[i]))
{
si.push(A[i]);
}
else
{
ss.insert(A[i]);
}
}
while(si.size())
{
int a=si.top(),b=a+1;
si.pop();
while(ss.count(b))
b++;
ss.insert(b);
count+=b-a;
}
return count;
}
};
2. 优秀算法
直观的感觉是:就像平整土地,我是讲所有的多余的土集中起来,然后哪里有坑哪里填;
优秀算法就是找平后向后退,多余的土地如果在随后就有坑,随后就填,不用集中。效率更好。
我自己的优秀方法,像推土机。
class Solution {
public:
int minIncrementForUnique(vector<int>& A) {
int N=A.size();
if(N<=1) return 0;
int count=0,t;
sort(A.begin(),A.end());
int r=A[0];
for(int i=1; i<N; i++) {
if(A[i]<=r) {
t=A[i];
r=r+1;
count+=(r-t);
} else r=A[i]; //不要忘了
}
return count;
}
};