C++——并查集

最新推荐文章于 2022-12-17 22:14:45 发布

东方旅行者

最新推荐文章于 2022-12-17 22:14:45 发布

阅读量145

点赞数

分类专栏：算法 C++ 文章标签：并查集 c++

本文链接：https://blog.csdn.net/qq_43270828/article/details/107073042

版权

算法同时被 2 个专栏收录

61 篇文章 3 订阅

订阅专栏

C++

25 篇文章 1 订阅

订阅专栏

直接上代码

代码

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
int  pre[1050];
//带路径压缩
int find(int x){
	int r=x;
	while(r!=pre[r])    r=pre[r];
	int i=x,j;
	while(pre[i]!=r){
		j=pre[i];
		pre[i]=r;
		i=j;
	}
	return r;
}
void join(int x,int y){
	int fx=find(x),fy=find(y);
	if(fx!=fy){
		if(fx<fy)   pre[fy]=fx;
		else        pre[fx]=fy;
	}
}
int main(){
	int num,edgeNum;
	cin>>num>>edgeNum;
	for(int i=0;i<num;i++)  pre[i]=i;
	while(edgeNum--){
        int a,b;
        cin>>a>>b;
        join(a,b);
	}
	for(int i=0;i<num;i++)
        cout<<find(i)<<endl;
        /*如果用pre[i]的话，
        可能还有部分结点在结束循环时还未进行路径压缩
        所以如果要找到一个结点的根父结点最好使用find函数
        */
	return 0;
}
/*
测试数据：
5 4
0 1
0 3
2 4
0 4
结果：
0
0
0
0
0
如果最后打印使用pre[i]的话，输出就是
0
0
0
0
2
*/