[蓝桥杯][2019年第十届真题]修改数组

题目描述
给定一个长度为 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。

样例输入

5
2 1 1 3 4

样例输出

2 1 3 4 5

---

方法一
思想比较直观，我最开始就是用这种方法做的，但是没有优化所以超时了，这是更改后的代码

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
int N;
int a[100005];
int book[1000006];

int main()
{
	memset(book,0,sizeof(book));
	cin>>N;	
	for(int i=0;i<N;i++){
		scanf("%d",&a[i]);
		if(book[a[i]]==0)book[a[i]]=1;
		else{
			int k=a[i];
			while(book[k]!=0){
				book[k]++;
				k+=book[k]-1;
			} 
			a[i]=k;
			book[k]=1;
		}
	}

	for(int i=0;i<N;i++){
		printf("%d ",a[i]);
	}
	return 0;
}

参考文章

---

方法二
用并查集，这种方法比较快

#include<iostream>
#include<algorithm>
using namespace std;
int a[100005];
int b[1000006];
int N; 
void init(){
	for(int i=0;i<1000006;i++)
		b[i]=i;
}
int getf(int t){
	if(b[t]==t)return t;
	else{
		b[t]=getf(b[t]);
		return b[t];
	}
}
void merge(int x,int y){
	int xf,yf;
	xf=getf(x);
	yf=getf(y);
	if(xf!=yf){
		b[xf]=yf;
	}
}
int main()
{
	init();
	cin>>N;
	for(int i=0;i<N;i++){
		scanf("%d",&a[i]);
		a[i]=getf(a[i]);
		merge(a[i],a[i]+1);
	}
	for(int i=0;i<N;i++)
	cout<<a[i]<<' ';
	cout<<endl;
	return 0;
}