对一堆数字进行并查集的基本操作:并+查
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5;
int p[N];
int find(int x){
if(p[x]!=x) return p[x]=find(p[x]);
return p[x];//写 return x 是一样的
}
int main(){
int n,m;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;++i) p[i]=i;
char op[2];
while(m--){
int a,b;
scanf("%s%d%d",&op,&a,&b);
if(op[0]=='M'){
p[find(a)]=find(b);
}else{
if(find(b)==find(a)) puts("Yes");
else puts("No");
}
}
}
算法竞赛进阶指南上find函数这样写
int find(int x){
if(x==p[x]) return x;
return p[x]=find(p[x]);
}
压行可以这么写
int find(int x){return p[x]==x?x:p[x]=find(p[x]);}
记忆化搜索也是这么写的(我是说return的时候赋值这种写法)