并查集可以简单的分为三个步骤:
1,init 初始化
2,find 查找
3,unionn 合并
1,init 首先让每个 节点的祖宗节点指向自己。
void init(int n){
for(int i=0;i<n;i++){
f[i]=i;
}
}
2,find 查找返回祖宗节点。
int find(int i){
if(f[i]==i)
return i;
else{
f[i]=find(f[i]);
return f[i];
//返回父节点
}
}
3,合并两个节点(这两个节点是 ‘亲戚’)的祖宗。
void unionn(int i,int j){
int fa_i = find(i);
int fa_b = find(j);
f[fa_i]=fa_b;
//因为 i 和 j是认识的,所以合并他们的祖先
}
调用顺序 init->unionn , 合并完后才可查找。
举个例子:4和3是亲戚, 3和2是亲戚,2和1是亲戚,求4和2是否是亲戚。
输入第一行为三个整数,n、m和q。n为人数;m为输入亲戚的对数;q为查询数。接下来m行,每行2个整数a和b,表示a和b是亲戚。接下来q行,每行2个整数c和d,即查询c和d是否是亲戚。
#include<bits/stdc++.h>
using namespace std;
#define MAXN 6000
int f[MAXN], hight[MAXN];
void init(int n){
for(int i=0;i<n;i++){
f[i]=i;
}
}
int find(int i){
if(f[i]==i)
return i;
else{
f[i]=find(f[i]);
return f[i];
//返回父节点
}
}
void unionn(int i,int j){
int fa_i = find(i);
int fa_b = find(j);
f[fa_i]=fa_b;
//因为 i 和 j是认识的,所以合并他们的祖先
}
int main(){
int n,m,q,x,y;
cin>>n>>m>>q;
init(n);
for(int i=0;i<m;i++){
cin>>x>>y;
unionn(x,y);
}
for(int i=0;i<q;i++){
cin>>x>>y;
if(find(x)==find(y))
cout<<"是";
else
cout<<"不是";
}
return 0;
}