并查集问题扩展 poj 1703

问题链接   点击打开链接

这是一道典型的带种类的并查集问题，我们开一个一维数组mark[N]来保存每一个节点与根节点的关系，0表示同类，1表示不同类。

初始化：

for(int i=1;i<=n;i++)
      {
          father[i]=i;
          mark[i]=0; //mark值为0
      }

查找函数：

int find(int x)
{
    if(x!=father[x])
    {
        int temp=father[x];   //用temp来保存路径上的每一个父节点
        father[x]=find(father[x]);    
        mark[x]=(mark[x]+mark[temp])%2;  //更新节点的mark值，%2以防止mark值大于1
    }
    return father[x];
}

向量偏移：x->r=x->xx+xx->r=mark[x]+mark[temp]

归并函数

void join(int x,int y)
{
    int xx=find(x);
    int yy=find(y);
    if(xx!=yy)
    {
        father[xx]=yy;
        mark[xx]=(mark[y]-mark[x]+1)%2;   //更新节点的mark值，%2以防止mark值大于1
    }
}

向量偏移：x->xx(mark[x])  y->yy(mark[y])  x->y(1)

因此有  xx->yy=xx->x+x->y+y->yy=mark[y]-mark[x]+1

上代码：

#include<stdio.h>
using namespace std;
int t,n,m;
char ad;
int a,b;
int father[100010];
int mark[100010];
int find(int x)
{
    if(x!=father[x])
    {
        int temp=father[x];
        father[x]=find(father[x]);
        mark[x]=(mark[x]+mark[temp])%2;
    }
    return father[x];
}
void join(int x,int y)
{
    int xx=find(x);
    int yy=find(y);
    if(xx!=yy)
    {
        father[xx]=yy;
        mark[xx]=(mark[y]-mark[x]+1)%2;
    }
}
int main()
{
  scanf("%d",&t);
  while(t--)
  {
      scanf("%d%d",&n,&m);
      getchar();
      for(int i=1;i<=n;i++)
      {
          father[i]=i;
          mark[i]=0;
      }
      while(m--)
      {
         scanf("%c%d%d",&ad,&a,&b);
         getchar();
          if(ad=='D')
          {
              join(a,b);
          }
          else if(ad=='A')
          {
            int x=find(a);
            int y=find(b);
            if(x!=y)
               printf("Not sure yet.\n");
            else
            {
            if(mark[a]==mark[b])
                printf("In the same gang.\n");
            else
                printf("In different gangs.\n");
            }
          }
      }
  }
  return 0;
}