poj 1703 Find them, Catch them //并查集

再一次做此题，温习并查集。

rank[x]记录的是x与father[x]的关系，值为0表示与根节点一致，1表示不同。初始化时，都一致，值为0。

ps:代码敲得比第一次的搓...

#include<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;
#define	N 100005
int father[N],rank[N];
int n,m;
void init()
{
	for(int i=1;i<=n;i++) father[i]=i,rank[i]=0;
}
int find(int k)
{
	if(k==father[k]) return k;
	int s=father[k];
	int t=find(father[k]);
	rank[k]=(rank[k]+rank[s])%2;//通过与旧父节点的比较，更新此点的信息
	father[k]=t;
	return t;
}
void U(int a,int b,int ra,int rb)
{
	father[rb]=ra;
	rank[rb]=(rank[a]+rank[b]+1)%2;
}
int main()
{
	int t;
	cin>>t;
	while(t--)
	{
		scanf("%d%d",&n,&m);
		init();
		getchar();
		while(m--)
		{
			char t;
			int a,b;
			scanf("%c %d %d",&t,&a,&b);
			getchar();
			if(t=='A')
			{
				int k1=find(a),k2=find(b);
				if(k1!=k2)
					printf("Not sure yet.\n");
				else if(rank[a]!=rank[b]) printf("In different gangs.\n");
				else printf("In the same gang.\n");
			}
			else 
			{
				int k1=find(a),k2=find(b);
				if(k1!=k2) 
					U(a,b,k1,k2);
			}
		}
	}
	return 0;
}