数据结构-并查集

不相交集合（即并查集，disjoint set）一般有两种操作：（1）查找某元素属于哪个集合；（2）合并两个集合。最常用的数据结构是并查集的森林实现。也就是说，在森林中每棵树代表一个集合。用树根来标识一个集合。

  

开始时，每个元素就是一个集合

void InitSet(int n)

{

    int i;

    for(i=0;i<n;i++)

    {

            Father[i]=i;

        rank[i]=0;

    }

}

查找(FindRoot)：寻找某一个元素所在集合的根。使用路径压缩来优化该函数。

int FindRoot(int p)

{

    if (father[p]!=p)father[p]=Findroot(father[p]);

    return father[p];

}

合并(Union)：将集合S1与S2合并，即把S1的根的父亲设置为S2的根。定义rank作为合并的启发函数值，刚建立的新集合rank为0，当两个rank相同的集合合并时，随便选一棵树拥有新根，并把它的rank加1；否则rank大的树拥有新根(启发式合并)。

void Union(int p,int q)

{

    int a,b;

    a=Find(p); b=Find(q);

if(rank[a]>rank[b])father[b]=a;

    else if(rank[a]<rank[b])father[a]=b;

    else

        {

        father[b]=a;

            rank[a]++;

    }

}

 

并查集模板：

int set[MAXN],rank[MAXN];

int FindSet(int x)

{

    if(set[x]!=x)

        set[x]=FindSet(set[x]);

    return set[x];

}

void MakeSet(int x)

{

    set[x]=x;

    rank[x]=0;

}

void Link(int a,int b)

{

    if(rank[a]>rank[b])

        set[b]=a;

    else if(rank[a]<rank[b])

        set[a]=b;

    else

    {

        set[a]=b;

        rank[b]++;

    }

}

void Union(int a,int b)

{

    Link(FindSet(a),FindSet(b));

}

pku-2236（并查集）

#include <stdio.h>
#include <string.h>

#define MAXN 1010

struct node
{
    int x,y;
};
node po[1010];
int set[MAXN],rank[MAXN],d2[MAXN][MAXN],op[MAXN];

//并查集
int FindSet(int x)
{
    if(set[x]!=x)
        set[x]=FindSet(set[x]);
    return set[x];
}
void MakeSet(int x)
{
    set[x]=x;
    rank[x]=0;
}
void Link(int a,int b)
{
    if(rank[a]>rank[b])
    { set[b]=a;  rank[a]++;}
    else if(rank[a]<rank[b])
    { set[a]=b; rank[b]++;}
    else 
    {
        set[a]=b;
        rank[b]++;
    }
}
void Union(int a,int b)
{
    Link(FindSet(a),FindSet(b));
}

int main()
{
    int n,i,j,d,a,b;
    char ch;
    //freopen("1.txt","r",stdin);
     
      scanf("%d%d",&n,&d);
     
      memset(op,0,sizeof(op));
      for(i=1;i<=n;i++)
      {
          MakeSet(i);
          scanf("%d%d",&po[i].x,&po[i].y);
      }
       for(i=1;i<=n;i++)
       {
           d2[i][i] = 0;
           for(j=1;j<=n;j++)
           {
               d2[j][i] = (po[i].x-po[j].x)*(po[i].x-po[j].x)+(po[i].y-po[j].y)*(po[i].y-po[j].y);
                d2[i][j] = d2[j][i];
           }
       }
       while(scanf("%s",&ch)!=EOF)
       {
           if(ch=='O') 
           {
               scanf("%d",&a);
               op[a] = 1;
               for(i=1;i<=n;i++)
                   if(i!=a&&op[i]==1&&(FindSet(i)==FindSet(a)||d2[i][a]<=d*d))
                     Union(i,a);
           }
           if(ch=='S')
           {
               scanf("%d%d",&a,&b);
                 if(FindSet(a)==FindSet(b))
                     printf("SUCCESS ");
                 else
                     printf("FAIL ");
           }
       }
    return 0;
}