并查集(模板)

int p[MAX];//父亲
int rank[MAX];//树的高度

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

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

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