POJ 1703 Find them, Catch them

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

应该是书上例题(食物链 POJ1822)的简单版

这里同样用书上很巧妙的思路

如何表达 在集合A 和 集合B 中

并不是 让i.set = A  类似于这样活着P[i] = A

而是将 这个变成一个隐形条件 并且 不用再关心 i到底是在A集合 还是在B集合 

建立并查集par[2*N] 

让i 表示在集合A 中 i +N 表示在集合B 中

当两个数在同一集合(有共同根节点)时 说明这样的时间必然发生

及如果same(i, j+N) 则说明 i 在集合A中 j在集合B中 --->>>>i j 一定不再同一个集合中

这样简化问题 不再纠结于 i到底是在A 还是在B 而是针对问题 操作 

注意不要用cin 会超时

#include <iostream>
#include <stdio.h>
#include <string.h>
#define N 100007
using namespace std;

int par[2*N];
int rank[2*N];
int find(int x)
{
    if (x == par[x]) return x;
    else return par[x] = find(par[x]);
}

void unite(int x, int y)
{
    int px = find(x), py = find(y);
    if (px == py) return ;
    else if (rank[px] > rank[py])
    {
        par[py] = px;
    }
    else
    {
        par[px] = py;
        if (rank[px] == rank[py]) rank[py]++;
    }
}

bool same(int x, int y)
{
    int px = find(x), py = find(y);
    return px == py;
}
int main()
{
    int T, n, M;
    freopen("in.txt", "r", stdin);
    scanf("%d", &T);
    while(T--)
    {
        scanf("%d%d", &n, &M);
        for (int i = 1;i <= 2*n; i++)
        {
            par[i] = i;
            rank[i] = 1;
        }
        char ch;
        int a, b;
        for (int i = 0; i < M; i++)
        {
            //cin >> ch;//用cin 读取 char 可以避免换行符 但是会超时 因为 输入规模比较大
            getchar();
            scanf("%c", &ch);
            scanf("%d%d", &a, &b);
            if (ch == 'A')
            {
                if ( same(a, b) || same(a, b+n) )
                {
                    if ( same(a, b) ) printf("In the same gang.
");
                    else printf("In different gangs.
");
                }
                else printf("Not sure yet.
");
            }
            else
            {
                unite(a, b+n);
                unite(a+n, b);
            }
        }
    }
    return 0;
}