修改第十届蓝桥杯省赛C/C++A组——试题 H: 修改数组(并查集应用)

修改数组
时间限制: 1.0s 内存限制: 256.0MB 本题总分:20 分
【问题描述】
给定一个长度为 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
【评测用例规模与约定】
对于 80% 的评测用例,1 ≤ N ≤ 10000。
对于所有评测用例,1 ≤ N ≤ 100000,1 ≤ Ai ≤ 1000000。

解题思路:

我们用并查集的思想来解决,我们初始化f数组为它本身。
从头到尾扫描给定的数组,遇到一个数,我们直接把这个数变成他的父亲。然后把这个结果与这个结果+1两个merge一下。为什么这样呢?我们来看个例子
1 4 5 4 6
扫第一个 用这个数的父亲来代替这个数,find(1)=1;
然后1 和 2 merge 这样如果再扫到 1 的话就直接跳过1 直接变成2了
1 2—>2
1 4 5 4 6
扫第二个 find(4)= 4
merge(4,5)
1 2—>2
4 5—>5
1 4 5 4 6
扫第三个 find(5) = 5
megre(5,6)
1 2 —>2
4 5 6—>6
扫第四个 find(4)=6
merge(6,7)
1 2—>2
4 5 6 7—>7

扫第五个 find(6)=7
merge(7,8)
1 2—>2
4 5 6 7 8—>8

这样,我们每次都是一个find()查询和一个merge()合并,并查集的查询和合并都是log级的,比暴力搜的复杂度要低了很多
(因为也没找到地方能提交代码 也不知道写的对不对 不过,,应该是对的).

代码:

#include 
#include 
#include 
using namespace std;
const int maxn=1e5+10;
int f[maxn];
int a[maxn];
int n;
int find(int x){
	if(x==f[x]){
		return f[x];
	}
	else{
		return f[x]=find(f[x]);
	}
}
void merge(int x,int y){
	int fx=find(x);
	int fy=find(y);
	if(fx==fy) return;
	else if(fx<fy){
		f[fx]=fy;
	}
	else{
		f[fy]=fx;
	}
	return ;
}
void init(){
	for(int i=1;i<=n;i++){
		f[i]=i;
	}
}
int main()
{

	cin>>n;
	init();
	map<int,int> mp;
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i]);
	}
	for(int i=1;i<=n;i++){
		int ans=find(a[i]);
		merge(ans,ans+1);
		cout<<ans;
		if(i==n) cout<<endl;
		else cout<<" ";
	}
	return 0;
}

你可能感兴趣的:(并查集)