并查集有两种操作————一是将两个元素合并,二是将找到元素的代表元素。
合并这项操作需要先找到两个元素的代表元素才能进行合并。 所以合并是以查找我=为基础的。而查找,可以通过压缩路径的方法来减少时间复杂度,来提高效率。
压缩路径其实就是将这些子节点指向代表元素。这项操作可以通过查找来实现,即边查找边压缩。
代码如下:
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
using namespace std;
const int maxn=1005;
int a[maxn];
int n,m;
//初始化
void init ()
{
for (int i=1;i<=n;i++)
a[i]=i;
}
//找到集合的代表
int finds (int x)
{
int temp;
if(x==a[x])
return x;
else
temp=finds(a[x]);
//路径压缩
a[x]=temp;
return temp;
}
//将两个集合联合起来
int unite (int x,int y)
{
int par1=finds(x);
int par2=finds(y);
if(par1!=par2)
a[par1]=par2;
}
int main()
{
scanf("%d%d",&n,&m);
init();
for (int i=0;i<m;i++)
{
int x,y;
scanf("%d%d",&x,&y);
//将两个元素所在的集合合并
unite(x,y);
}
return 0;
}