Find them, Catch them(带权并查集(裸))

题目来源: https://vjudge.net/problem/POJ-1703
【题意】
大致意思是一座城市里有两大帮派,召集了n个人,然后输入的时候,若是D,x,y,给出一种关系,x和y属于不同的阵营。若是A x y,询问你x和y的关系。
【思路】
就像食物链(http://blog.csdn.net/duan_1998/article/details/70211421)那道题一样,找出节点与根节点的关系,若是有不同的祖先,那就把一个祖先赋给另一个祖先当父节点,然后更新当儿子的祖先的所有子节点的对应关系。(若是不懂,看下食物链的链接)
【代码】

#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;
typedef long long LL;
int pre[100000+10];
int ranks[100000+10];
int find(int x)
{
    if(x!=pre[x])
    {
        int fx=pre[x];
        pre[x]=find(pre[x]);
        ranks[x]=(ranks[fx]+ranks[x])&1;
    }
    return pre[x];
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n,m;
        scanf("%d%d%*c",&n,&m);
        for(int i=1; i<=n; i++)
            pre[i]=i,ranks[i]=0;
        for(int i=0; ichar s;
            int a,b;
            scanf("%c%d%d%*c",&s,&a,&b);
            int fa=find(a);
            int fb=find(b);
            if(s=='D')
            {
                if(fa!=fb)
                    pre[fb]=fa;
                ranks[fb]=(ranks[a]-ranks[b]+1)&1;
            }
            else if(s=='A')
            {
                if(fa!=fb)
                    printf("Not sure yet.\n");
                else if(fa==fb&&ranks[a]!=ranks[b])
                    printf("In different gangs.\n");
                else if(fa==fb&&ranks[a]==ranks[b])
                    printf("In the same gang.\n");
            }
        }
    }
}

你可能感兴趣的:(ACM竞赛,【数据结构】--并查集,ACM的进程)