1242 修改数组（第十届蓝桥杯）

给定一个长度为 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;
 }