什么是并查集
并查集是类似于集合的一种关系
主要的基本用法有两个:
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;
}
希望对你有帮助!