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]);}
記憶化検索もこのように書いています.