【POJ 1703 Find them, Catch them】并查集

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

题目大意：

给定m个操作：

D x y：x罪犯和y罪犯不在一个团伙。（总共只有两个团伙）

A x y：询问x和y罪犯的关系

 

解题思路：

    很容易想到是并查集，但最后还是让我纠结了良久。一直MLE。

    对于每次的D x y操作，每次不仅要找到两个不同集合标志头进行更新，还要更新a数组，即存储的相对集合的标志头。

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

const int maxn=100005;
int f[maxn], a[maxn];  ///a数组表示的是这个集合指向另一个相对的集合

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

int main()
{
    int T, n, m;
    cin >> T;
    while(T--)
    {
        cin >> n >> m;
        for(int i=0; i<=n; i++)
            f[i]=i, a[i]=0;
        char ch[2];
        int x, y;
        while(m--)
        {
            scanf("%s%d%d",ch,&x,&y);
            if(ch[0]=='D')
            {
                x=find(x); ///注意这里指向两个不同集合的标志头
                y=find(y);
                if(!a[x]) a[x]=y;
                if(!a[y]) a[y]=x;
                a[x]=find(a[x]);///!!!这里也要注意更新，指向另一集合的标志头
                a[y]=find(a[y]);
                f[a[x]]=y;
                f[a[y]]=x;
            }
            else
            {
                x=find(x),y=find(y);
                if(x==y)
                    puts("In the same gang.");
                else if(find(a[x])==y)
                    puts("In different gangs.");
                else
                    puts("Not sure yet.");
            }
        }
    }
    return 0;
}