【POJ】1703 - Find them, Catch them

http://poj.org/problem?id=1703

N名来自两个帮派的坏东西。
2种操作：
（1）：给定2个坏东西不属于同一帮派。
（2）：判断2个坏东西是否属于同一帮派。

并查集。
设元素个数为2 * N，i表示属于帮派A，i + N表示属于帮派B，每次输入两个家伙不在同一帮派的时候，就合并他们分属两个帮派的元素。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <queue>
#include <algorithm>
using namespace std;

const int maxn=100000*2+5;

int t;
int n,m;
int fa[maxn];
int rank[maxn];

void init(){
    for (int i=0;i<n*2;i++){
        fa[i]=i;
        rank[i]=0;
    }
}

int find(int x){
    if (x==fa[x]) return x;

    return fa[x]=find(fa[x]);
}

void unite(int x,int y){
    x=find(x);
    y=find(y);

    if (x==y) return;

    if (rank[x]<rank[y]){
        fa[x]=y;
    }
    else{
        fa[y]=x;
        if (rank[x]==rank[y]){
            rank[x]++;
        }
    }
}

bool same(int x,int y){
    return find(x)==find(y);
}
int main(){

    cin >> t;
    while (t--){
        cin >> n >> m;
        init();
        getchar();
        for (int i=0;i<m;i++){
            char ch;
            int x,y;
            scanf("%c%d%d",&ch,&x,&y);
            getchar();
            if (ch=='A'){
                if (same(x,y)){
                    cout << "In the same gang." << endl;
                }
                else
                if (same(x,y+n)){
                    cout << "In different gangs." << endl;
                }
                else{
                    cout << "Not sure yet." << endl;
                }
            }
            else{
                unite(x,y+n);
                unite(x+n,y);
            }
        }
    }
}