yp 极简 互质的集合 Disjoint Set：Union Find Tree C++

直接代码

#include<bits/stdc++.h>
using namespace std;

class DisjoinSet{
public:
    vector<int> rank,p;
    
    DisjoinSet(){}
    DisjoinSet(int size){
        rank.resize(size);
        p.resize(size);
        for(int i = 0;i < size;i++) makeSet(i);
    }
/*初始化*/
    void makeSet(int x){
        p[x] = x;
        rank[x] = 0;
    }

/*判断两个数是否为一个集合*/
    bool same(int x,int y){
        return findSet(x) == findSet(y);
    }
/*合并标准化接口*/
    void unite(int x,int y){
        link(findSet(x),findSet(y));//合并两个代表元素
    }
/*合并集合*/
    void link(int x,int y){
        if(rank[x] > rank[y]){//为了尽量减少树的高度，合并的时候，将低的合并到高的处
            p[y] = x;
        }else {
            p[x] = y;
            if(rank[x] == rank[y]){//合并两颗高度相同的树，代表元素的高度加一
                rank[y]++;
            }
        }
    }

/*寻找x的终极父节点*/
    int findSet(int x){
        if(x != p[x]){
            p[x] = findSet(p[x]);
        }
        return p[x];
    }
};

int main(){
    int n,a,b,q;
    int t;

    cin>>n>>q;
    DisjoinSet ds = DisjoinSet(n);

    for(int i=0;i<q;i++){
        cin>>t>>a>>b;
        if(t == 0) ds.unite(a,b);
        else if(t == 1){
            if(ds.same(a,b)) cout<<1<<endl;
            else cout<<0<<endl;
        }
    }

    system("pause");
    return 0;
}