POJ1703Find them, Catch them 【种类并查集】

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

题目大意：给n个人,m次询问。A代表询问a, b之间的关系，D代表给出a, b属于不同的帮派。

我的想法：

太菜了，上课的时候没想到这种方法。

思路：

1.感觉有点像2-sat的思想，对于D给出的a, b两人属于不同帮派。我们就将a于b的对立点用并查集维护在一个集合中，也将b于a的对立点连通。

2.查询时，若a,b两人pre值相等则属于同一帮派。若a与b的对立点属于同一集合，或b与a的对立点属于同一集合，则说明a与b属于不同帮派。若上述都不成立则说明关系还未知。

3.对立点是虚构的，不能影响到原来的点。所以数据范围翻倍，x + n代表x的对立点。

注意：不能用 cin 输入,会超时。只能用scanf

#include<stdio.h>
const int maxn = 2e5 + 1e4;

int pre[maxn];

int find(int x)
{
    if(pre[x] == x)
        return x;
    else
    {
        int root = find(pre[x]);
        pre[x] = root;
        return pre[x];
    }
}

int main()
{
    int T;
    scanf("%d", &T);
    while(T --)
    {
        int n, m;
        scanf("%d%d", &n, &m);
        getchar();
        for(int i = 1; i <= 2 * n; i ++)
        {
            pre[i] = i;
        }
        for(int i = 1; i <= m; i ++)
        {
            char ch;
            int a, b;
            scanf("%c%d%d", &ch, &a, &b);
            if(ch == 'D')
            {
                int x = find(a), y = find(b);
                int xx = find(a + n), yy = find(b + n);
                if(x != yy)
                    pre[yy] = x;
                if(y != xx)
                    pre[xx] = y;
            }
            if(ch == 'A')
            {
                int x = find(a), y = find(b);
                int xx = find(a + n), yy = find(b + n);
                if(x == y)
                    printf("In the same gang.
");
                else if(x == yy || y == xx)
                    printf("In different gangs.
");
                else
                    printf("Not sure yet.
");
            }
            getchar();
        }
    }
    return 0;
}