PS:B站上大佬的讲解
路径压缩后的并查集查找函数:
int FindFather(int v){
if(v == father[v]) return v; //如果节点V的父亲节点为自身,则返回自身节点
else{
//递归找到父亲节点,并将路径上的节点的父亲节点全部设置为总的父亲节点。
int F = FindFather(father[v]);
father[v] = F;
return F;
}
}
集合的合并函数:
void Union(int a,int b){
int fa = FindFather(a); //查找a的父亲节点,并将其设为fa
int fb = FindFather(b); //查找b的父亲节点,并将其设为fb
if(fa != fb){
father[fa] = fb; //如果两个父亲节点不相等,则将两个父亲节点合并
//注意:并非将两个节点合并,而是将两个节点的父亲节点合并
}
}
例子:
题目描述:
有一个叫做“数码世界”奇异空间,在数码世界里生活着许许多多的数码宝贝,其中有些数码宝贝之间可能是好朋友,并且数码宝贝世界有两条不成文的规定:
第一,数码宝贝A和数码宝贝B是好朋友等价于数码宝贝B与数码宝贝A是好朋友
第二,如果数码宝贝A和数码宝贝C是好朋友,而数码宝贝B和数码宝贝C也是好朋友,那么A和B也是好朋友,现在给出这些数码宝贝中所有好朋友的信息,问:可以把这些数码宝贝分成多少组,满足每组中的任意两个数码宝贝都是好朋友,而且任意两组之间的数码宝贝都不是好朋友
输入格式:
输入的第一行给右两个正整数n(n<=100)和m(m<=100),分别表示数码宝贝的个数和好朋友的组数,其中数码宝贝编号为1~n。
接下来有m行,每行两个正整数a和b,表示数码宝贝a和数码宝贝b是好朋友。
输出格式:
输出一个整数,表示这些数码宝贝可以分成的组数
样例输入:
4 2
1 4
2 3
样例输出:
2
代码:
#include<bits/stdc++.h>
using namespace std;
int n,m,father[100],re = 0;
int FindFather(int v){ //路径压缩,寻找根节点
//一. 如果节点的父亲节点为自身,退出
if(father[v] == v) return v;
else{
//二. 递归找到父亲节点
int F = FindFather(father[v]);
//三. 对路径上的节点的父亲节点都设为F
father[v] = F;
return F;
}
}
void Union(int a,int b){ //合并两个集合
//一. 寻找两个节点的父亲节点
int fa = FindFather(a);
int fb = FindFather(b);
//二. 如果非同一个父亲节点,则合并两个父亲节点
if(fa != fb){
father[fa] = fb;
}
}
int main(){
cin>>n>>m;
int a,b;
for(int i = 0;i<n;i++){
//初始化father[]
father[i] = i;
}
for(int i = 0;i<m;i++){
cin>>a>>b;
Union(a,b);
//合并两个节点所在的集合
}
for(int i = 0;i<n;i++){
//查询根节点的数目
if(father[i] == i)
re++;
}
cout<<re<<endl;
return 0;
}
end.