POJ1703--Find them, Catch them

题目大意：有N个罪犯，分属于两个团伙。有两种操作，一、D表示两个罪犯不在同一个团伙，二、A表示判断两个罪犯是否属于同一个团伙。操作m次

 

分析：对每个罪犯创建2个元素，i-x,i-y，都用并查集维护。i-x表示i属于x类。查询的时候判断i-x与j-x，i-x与j-y的关系即可。合并的时候，合并i-x与j-y，j-x与i-y。

代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int maxn = 111111;

int n, m;
int fa[2*maxn+1];

int Find(int x) {
return fa[x] == x ? x : fa[x] = Find(fa[x]);
}

void Union(int x, int y) {
int p1 = Find(x);
int p2 = Find(y);
if(p1 != p2)
fa[p1] = p2;
}

int main() {
int T;
scanf("%d", &T);
while(T--) {
scanf("%d%d", &n, &m);
for(int i = 1; i <= n; i++) {
fa[i] = i;
fa[n+i] = n+i;
}

char op[2];
while(m--) {
int a, b;
scanf("%s%d%d", op, &a, &b);
int p1 = Find(a);
int p2 = Find(b);
int p3 = Find(n+a);
int p4 = Find(n+b);
if(op[0] == 'A') {
if(p1 != p2 && p1 != p4)
printf("Not sure yet.\n");
else if(p1 != p2 || p3 != p4)
printf("In different gangs.\n");
else printf("In the same gang.\n");
}
else {
Union(a, b+n);
Union(a+n, b);
}

}
}

return 0;
}