poj 1703 Find them, Catch them（并查集）

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

题意：一个地方有两个帮派， 每个罪犯只属于其中一个帮派，D 后输入的是两个人属于不同的帮派，

A后询问 两个人是否属于 同一个帮派。

用op[]数组记录 不跟自己在一个帮派中的一个人， 然后再输入不跟自己在一个帮派的人的时候，把

这个人跟 op[]数组记录里的那个人联系起来， 这两个人属于一个帮派。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
using namespace std;
const int maxn = 100010;
int bin[maxn], op[maxn], n;
void init()
{
    for(int i = 1; i <= n; i++)
    {
        bin[i] = i;
        op[i] = 0;
    }
}
int find(int x)
{
    int r, i, j;
    r =  x;
    while(r != bin[r])
    r = bin[r];
    i = x;
    while(bin[i] != r)
    {
        j = bin[i];
        bin[i] = r;
        i = j;
    }
    return r;
}
/*
int find(int x)  //递归 路径压缩写法
{
    return bin[x]==x?x:(bin[x]=find(bin[x]));
}
*/
void merge(int x, int y)
{
    int fx, fy;
    fx = find(x);
    fy = find(y);
    if(fx != fy)
    bin[fx] = fy;
}
int main()
{
    int t, m, a, b;
    char ch;
    scanf("%d", &t);
    while(t--)
    {
        scanf("%d%d", &n, &m);
        init();

        while(m--)
        {
           getchar();
           scanf("%c%d%d",&ch, &a, &b);
           if(ch == 'A')
           {
               int x, y;
               x = find(a);
               y = find(b);
               if(x == y)
               printf("In the same gang.
");
               else if(op[a]!=0&&find(op[a])==y || op[b]!=0&&find(op[b])==x)
               printf("In different gangs.
");
               else
               printf("Not sure yet.
");
           }
           else
           {
               if(op[a] == 0)
               op[a] = b;
               else
               merge(op[a],b);
               if(op[b] == 0)
               op[b] =a;
               else
               merge(op[b],a);
           }
        }
    }
    return 0;
}