poj 1703 Find them, Catch them(Dsu)

题目链接：http://poj.org/problem?id=1703

用二维数组，第一维记录元素i（f[i][0]），第二维记录与i不在同一集合的元素j（f[i][1]）。

初始化f[i][0]=i,f[i][1]=0（表示没有元素）;

每次操作时，合并f[a][0]和f[b][1],f[a][1]和f[b][0]。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<math.h>
#include<algorithm>
using namespace std;

const int INF=0x3f3f3f3f;
const int maxn=100005;
int T,n,m;
int f[maxn][2];

void Make_Set(){
	for(int i=1;i<=n;i++)
		f[i][0]=i,f[i][1]=0;
}

int find(int x){
	return x==f[x][0]?x:f[x][0]=find(f[x][0]);
}

void Union(int a,int b){
	int ra=find(a);
	int rb=find(b);
	if(ra==rb) return ;
	f[ra][0]=rb;
}

int main() {
#ifndef ONLINE_JUDGE
	freopen("test.in","r",stdin);
	freopen("test.out","w",stdout);
#endif
	scanf("%d",&T);
	while(T--){
		scanf("%d%d",&n,&m);
		Make_Set();
		char ch;
		int a,b;
		while(m--){
			getchar();
			scanf("%c%d%d",&ch,&a,&b);
			int tmpa=f[a][1];
			int tmpb=f[b][1];
			if(ch=='D'){
				if(!tmpa) f[a][1]=tmpa=b;
				if(!tmpb) f[b][1]=tmpb=a;
				Union(a,tmpb),Union(tmpa,b);
			}
			if(ch=='A'){
				if(find(f[a][0])==find(f[b][0])) puts("In the same gang.");
				else if(find(f[a][1])==find(f[b][0])) puts("In different gangs.");
				else puts("Not sure yet.");
			}
		}
	}
	return 0;
}