ACWing 836. 合并集合 (模板题 并查集)

题目链接

对一堆数字进行并查集的基本操作：并+查

#include
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的时候赋值这种写法）