c++ 并查集

什么是并查集

并查集是类似于集合的一种关系

主要的基本用法有两个：
1、判断两个点是否属于同一个集合
2、查询现有的集合个数

算法思路

用故事讲算法：
江湖里来了九位英雄人物：张无忌、小昭、光明左使杨逍、白眉鹰王殷天正、金毛狮王谢逊、青翼蝠王韦一笑、灭绝师太、周芷若、成昆。
首先呢，给他们个取个代号
0：张无忌
1：小昭
2：光明左使杨逍
3：白眉鹰王殷天正
4：金毛狮王谢逊
5：青翼蝠王韦一笑
6：灭绝师太
7：周芷若
8：成昆

他们初来驾到，来到这里时都互不认识，他们一个人就代表一个势力

int f[9];
for(int i=0;i<9;i++){
	f[i]=i;//第i号认识的人就是i自己
}

张无忌在光明顶与小昭相识，可以这么认为，只要小昭承认认识张无忌，那么张无忌就默认认识小昭，因此只要让f[1]=0，就可以证明张无忌和小昭是认识的

f[1]=0;

怎么证明呢？江湖有句话，朋友的朋友就是朋友，小昭认识张无忌，张无忌也认识张无忌自己，那么张无忌和小昭就互相认识，现在的状态是：
小昭->张无忌->张无忌
张无忌->张无忌
他们寻找的最终目标都是一样的，因此我们可以用以下函数实现查询，直到查询到自己认识自己的人

int find(int a){
	if(a==f[a]){//自己认识自己
		return a;
	}else{
		return find(f[a]);//递归查找认识的人
	}
}

随后杨逍、殷天正、谢逊、韦一笑执掌了明教，他们的关系是这样的

f[3]=2;
f[4]=2;
f[5]=2;

谢逊没有直接和殷天正认识，怎么说明他们是认识的呢？谢逊说他大哥是杨逍，殷天正说他大哥也是杨逍，一拍即合，好兄弟！因此用以上函数就可以证明他们的关系了

if(find(3)==find(4)){
	cout<<"他们俩认识"<<endl; 
}else{
	cout<<"他们俩不认识"<<endl; 
}

以上说的都是从没有关系建立关系的，那么现在谢逊以及认杨逍为大哥了，他又想认识张无忌，那该怎么办呢？如果自己去认识他的话他就不能认杨逍为大个了，有个很好的办法就是让大哥杨逍去认识张无忌，杨逍认张无忌为大哥，那么杨逍全部小弟都会认张无忌为大哥

f[find(4)]=0;

假如谢逊想认识小昭也是同样的道理，让杨逍认识张无忌或者张无忌认识杨逍都可以

f[find(4)]=find(1);

我们用find()函数时，寻找到的结果都会是大哥张无忌，可以证明这个圈子里的人都是朋友，所以find()函数实际就是寻找大哥的函数

例如，灭绝师太和周芷若建立了关系

f[7]=6;

那么周芷若的大哥就是灭绝师太

怎么查询这个武林有多少方势力呢？我们只要知道这个武林有多少大哥就行了，可以发现，大哥都是认识自己的，也就是f[i]==i；其它当小弟的都会认识其它人，也就到此为止，只有三个大哥只认识自己，张无忌s[0]=0,灭绝师太f[6]=6,成昆f[8]=8，所以这个武林只有3方势力

int ans=0;
for(int i=0;i<9;i++){
	if(f[i]==i){
		ans++;
	}
}

优化
路径压缩：可以看到，谢逊在找张无忌的时候要先先找到杨逍，那么路径压缩要做的就是杨谢逊直接找到张无忌

int find(a){
	if(a==find(a)){
		return a;
	}else{
		return f[a]=find(f[a]);//让他认识的人直接等于大哥
	}
}

代码参考

#include<iostream>
using namespace std;
int f[9];
int find(int a){
	if(a==f[a]){
		return a;
	}else{
		return f[a]=find(f[a]);//让他认识的人直接等于大哥
	}
}
int main(){
	for(int i=0;i<9;i++){
		f[i]=i;
	}
	f[1]=0;//小昭认识张无忌 
	
	f[3]=2;//殷天正认识杨逍 
	f[4]=2;//谢逊认识杨逍 
	f[5]=2;//韦一笑认识杨逍 
	
	f[find(4)]=find(1);//让谢逊的大哥认识小昭的大哥 

	
	if(find(1)==find(4)){
		cout<<"小昭和谢逊认识"<<endl; 
	}else{
		cout<<"小昭和谢逊不认识"<<endl; 
	}
	f[7]=6;//周芷若认识灭绝师太 
	int ans=0;
	for(int i=0;i<9;i++){
		if(f[i]==i){
			ans++;
		}
	}
	cout<<"现在有 "<<ans<<' '<<"方势力" <<endl;
}

希望对你有帮助！