POJ 1703 Find them, Catch them 种类并查集

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <limits.h>
#include <math.h>
#include <iostream>
#include <algorithm>
using namespace std;
#define N 500000+5
int fa
,rank
;
void init(int n){//初始化
for(int i=0; i<=n; i++){
fa[i]=i;
rank[i]=0;
}
}
int find(int x)//找父节点
{
if(x==fa[x])    return x;
else
{
int f=fa[x];
fa[x]=find(fa[x]);//递归
rank[x]=(rank[x]+rank[f])%2;//更新rank值
return fa[x];
}
}
void merge(int x,int y)
{
int fx=find(x),fy=find(y);
if(fx==fy)  return ;//
else
{
fa[fy]=fx;  //更新fa
rank[fy]=(rank[x]-rank[y]+2+1)%2;
}
}
int main()
{
//   freopen("in.txt","r",stdin);
int n,m,t;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&m);
init(n);
int ans=0;
for(int i=0; i<m; i++)
{
int b,c;
char a;
getchar();
scanf("%c%d%d",&a,&b,&c);
if(a=='D')
merge(b,c);
else
{
int x=find(b),y=find(c);
if(y!=x)//没有相同的父节点表示先前没有联系
printf("Not sure yet.\n");
else if(rank[b]==rank[c]&&x==y)//父节点相同 并且rank值%2==0（0 +0或者1+1）
printf("In the same gang.\n");
else printf("In different gangs.\n");
}
}
}
return 0;
}