1.路径压缩+查找根节点
int find(int x){
if(p[x]!=x) p[x]=find(p[x]);
return p[x];
}
2.合并两个集合+记录连通块中元素数目
void merge(int a, int b){
if(find(a)==find(b)) return;
p[find(a)]=find(b);
cnt[find(b)]+=cnt[find[a]];
}
合并集合:https://www.acwing.com/problem/content/838/
#include<iostream>
using namespace std;
const int N=1e5+5;
int p[N];
//查找+路径压缩
int find(int x){
if(p[x]!=x) p[x]=find(p[x]);
return p[x];
}
int main(){
int n,m; cin>>n>>m;
for(int i=1;i<=n;i++) p[i]=i;
while(m--){
char op;
int a,b;
cin>>op>>a>>b;
if(op=='M'){
p[find(a)]=find(b);
}
else{
if(find(a)==find(b)) printf("Yes\n");
else printf("No\n");
}
}
return 0;
}
连通块中点的数目:https://www.acwing.com/problem/content/839/
#include<iostream>
#include<cstdio>
using namespace std;
const int N=1e5+5;
int p[N],cnt[N];
int find(int x){
if(p[x]!=x) p[x]=find(p[x]);
return p[x];
}
int main(){
int n,m;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
p[i]=i;
cnt[i]=1;
}
while(m--){
char op[5];
scanf("%s",op);
if(op[0]=='C'){
int a,b;
scanf("%d%d",&a,&b);
if(find(a)!=find(b))
cnt[find(b)]+=cnt[find(a)];
p[find(a)]=find(b);
}
else{
if(op[1]=='1'){
int a,b;
scanf("%d%d",&a,&b);
if(find(a)==find(b)) puts("Yes");
else puts("No");
}
else{
int a;
cin>>a;
printf("%d\n",cnt[find(a)]);
}
}
}
return 0;
}