蓝桥杯 修改数组(并查集)

蓝桥杯 修改数组(并查集)

 一开始没什么思路,暴力显然会T,挺难想到用并查集来做的

p[x]指向的就是x应该修改成的数,在find的过程中,其实就是找到第一个没有出现过的数

#include<iostream>
#include<algorithm>
#include<map>
using namespace std;
typedef long long ll;
const int N=1e6+5;
int p[N];//p[x]指向x改变后的数 
int find(int x)
{
	if(x!=p[x]) p[x]=find(p[x]);
	return p[x]; 
}
int a[N];
int main()
{
	int n;cin>>n;
	for(int i=1;i<=n;i++) cin>>a[i];
	for(int i=1;i<=N;i++) p[i]=i;
	for(int i=1;i<=n;i++)
	{
		a[i]=find(a[i]);
		p[a[i]]=find(a[i]+1);
		//因为a[i]在这里被用过了,所以之后的a[i]应该改变为p[a[i]+1]
		//即a[i]+1应该改变后的数(如果a[i]+1在之前也被用过,p[a[i]+1]指向的就是正确的数) 
	}
	for(int i=1;i<=n;i++)
	{
		cout<<a[i];
		if(i!=n) cout<<" ";
	}
}

上一篇:算法基础课——连通块中点的数量


下一篇:Java静态域