POj 1703 Find them, Catch them(关系并查集)

题目: 戳我

题意:城市中有两个社会团伙

已知有n个人：编号为1~n，和m个信息

对于每个信息，D a b表明a和b属于不同的团伙，并且信息不会相互矛盾；A a b表示讯问a和b的关系，

要求依次回答每个讯问，回答形式如下：

Not sure yet.       //a和b的关系不确定

In different gans.  //a和b属于不同的团伙

In the same gang. //a和b属于同一个团伙

回顾并查集的合并操作：我们总是将同类的集合进行合并，但此题给出的却是不同类的两个元素。

抓住关键，城市的团伙只有两个，这个信息告诉我们：如果a和b不同类，b和c不同类，那么a和c必然属于一个团伙。

更一般的，如果我们把编号看做顶点，关系看做连边。那么，如果顶点a,b之间的路径包含奇数条边，那么a与b属于不同的团伙，如果a,b之间的路径包含偶数条边，那么a与b属于同一个团伙。

我们能够很容易的计算出元素x与根节点的“距离”（即路径上经过的边数）dis[x]。类似的，对于与x同属一个集合的任意元素y，我们也可以计算出y与根节点的“距离”dis[y]。那么，x与y的“距离”便可以用dis[x]+dis[y]表示。（注意，我们只关心“距离”的奇偶性，并不关心“距离”是否是最短的）

代码:

#include <cstdio>

#include <cstdlib>

#include <cstring>

#include <cctype>

#include <cmath>

#include <algorithm>

#include <vector>

#include <queue>

#include <stack>

#include <map>



using namespace std;



const int inf = 0x3f;

const int INF = 0x3f3f3f3f;

const int maxn = 1e5+5;



int bleg[maxn], dis[maxn], n, q;



int find(int x)

{

    int y = x, cnt = 0;

    while(y != bleg[y])

    {

        cnt += dis[y];

        y = bleg[y];

    }

    while(x != bleg[x])

    {

        int px = bleg[x], tmp = dis[x];

        dis[x] = cnt;

        bleg[x] = y;

        cnt -= tmp;

        x = px;

    }

    return y;

}



void Union(int a, int b)

{

    int pa = find(a), pb = find(b);

    if(pa == pb) return;

    dis[pa] = dis[a] + dis[b] + 1;

    bleg[pa] = pb;

}



void Init()

{

    for(int i = 0; i <= n; i++)

    {

        bleg[i] = i; dis[i] = 0;

    }

}

int main()

{

    int T;

    scanf("%d", &T);

    while(T--)

    {

        scanf("%d %d", &n, &q);

        Init();

        while(q--)

        {

            int x, y;

            char op;

            scanf(" %c %d %d", &op, &x, &y);

            if(op == 'A')

            {

                int pa = find(x), pb = find(y);

                if(pa != pb)

                {

                    puts("Not sure yet.");

                }

                else if((dis[x] + dis[y]) % 2 != 0)

                {

                    puts("In different gangs.");

                    //printf("%d %d\n", dis[x], dis[y]);

                }

                else 

                {

                    puts("In the same gang.");

                }

            }

            else 

            {

                Union(x, y);

            }

        }

    }

    return 0;

}