并查集简单实现

const int maxn = 50010;
int root[maxn];//父亲
int rank[maxn];//数的高度
int n;

//初始化n个元素
void init(){
    for(int i = 0; i < n; i++){
        root[i] = i;
        rank[i] = 0;
    }
}

//查询树的根
int find(int x){
    if(root[x] = x) return x;
    return root[x] = find(root[x]);
}

//合并x和y所属的集合
void unit(int x, int y){
    x = find(x);
    y = find(y);
    if(x == y) return;
    if(rank[x] < rank[y]){
        root[x] = y;
    }else{
        root[y] = x;
        if(rank[x] == rank[y])
            rank[x]++;
    }
}

//判断x和y是否属于同一个集合
bool same(int x, int y){
    return find(x) == find(y);
}