并查集 数据结构

并查集（用树形结构实现的）

有2种操作：

（1）查询元素a和元素b是否在同一组。

（2）合并元素a和元素b所在的组。

int pre[N];//我们用数组下标代表每个元素，而数组pre[N]代表N父亲的编号，

int rank[N]; //rank[N]代表这个树的高度       // 当pre[N]=N时，N就是所在树的根
void init(void)
{
       for(i=0;i<n;i++)
       {
             pre[i]=i;
             rank[i]=1;
       }
}
int find(int x)//找根节点，如果pre[x]==x，那么x就是其根节点
{                                 //否则，找pre[x]的父亲节点也就是pre[pre[x]],看看是否等于x。。。
         return pre[x]==x?x:find(pre[x]);
}
void unite(int x,int y)
{
        int root1,root2;
        root1=find(x);//找到2个元素各自的根节点root1,root2
        root2=find(y);
        if(root1==root2)//如果根节点相同，那就不用合并
             return ;
        if(rank[root1]<rank[root2])//b=比较2个树的高度，把小树合并到大树上
             pre[root1]=root2;//把小树的根节点的父亲节点指向大树的根节点root2
        else
        {
              pre[root2]=root1;
              if(rank[root1]==rank[root2])//如果2棵树高度相同，合并之后，高度会增加一
              rank[root1]++;
        }
}