给定一个长度为 N 的数组 A=[A1,A2,⋅⋅⋅AN],数组中有可能有重复出现的整数。
现在小明要按以下方法将其修改为没有重复整数的数组。
小明会依次修改 A2,A3,⋅⋅⋅,AN。
当修改 Ai 时,小明会检查 Ai 是否在 A1∼Ai−1 中出现过。
如果出现过,则小明会给 Ai 加上 1;如果新的 Ai 仍在之前出现过,小明会持续给 Ai 加 1,直到 Ai 没有在 A1∼Ai−1 中出现过。
当 AN 也经过上述修改之后,显然 A 数组中就没有重复的整数了。
现在给定初始的 A 数组,请你计算出最终的 A 数组。
输入格式
第一行包含一个整数 N。
第二行包含 N 个整数 A1,A2,⋅⋅⋅,AN。
输出格式
输出 N 个整数,依次是最终的 A1,A2,⋅⋅⋅,AN。
数据范围
1≤N≤105,
1≤Ai≤106
输入样例:
5
2 1 1 3 4
输出样例:
2 1 3 4 5
能过一部分数据。
#include<iostream>
#include<algorithm>
#include<map>
using namespace std;
int a[100007];
map<int,bool> mp;//unordered_map 也过不了
int main()
{
int n;
scanf("%d",&n);
for(int i=0;i<n;i++)
{
scanf("%d",&a[i]);
int x=a[i];
if(mp.find(x)==mp.end())
{
mp[x]=true;
}
else
{
while( mp.find(x)!=mp.end() )
{
x++;
}
mp[x]=true;
a[i]=x;
}
}
for(int i=0;i<n;i++)
cout<<a[i]<<" ";
return 0;
}
正解: 并查集(压缩路径)
#include<iostream>
#include<algorithm>
using namespace std;
int a[100007];
int fa[1000007];
int find(int x)
{
if(x!=fa[x])fa[x]=find(fa[x]);//压缩路径
return fa[x];
}
int main()
{
int n;
scanf("%d",&n);
for(int i=0;i<1000007;i++)
fa[i]=i;
for(int i=0;i<n;i++)
{
scanf("%d",&a[i]);
int x=find( a[i] );
printf("%d ",x);
fa[x]=x+1;
}
return 0;
}