并查集问题扩展 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}
}

因此有  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");elseprintf("In different gangs.\n");}}}}return 0;
}