poj1703 Find them, Catch them（带权并查集）

题目链接

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

题意

有两个帮派：龙帮和蛇帮，两个帮派共有n个人（编号1~n），输入m组数据，每组数据为D [a][b]或A [a][b]，D[a][b]表示a，b属于不同的帮派，A [a][b]则让我们判断a，b是否属于一个帮派，根据判断的结果进行相应的输出。

思路

这题和poj2492很像，使用并查集解决，方法我已在poj2492的题解中写出，这里不再赘述。

代码

#include <cstdio>
using namespace std;

const int N = 100000 + 10;
int p[N];
int r[N];

void make_set(int n)
{
    for (int i = 1;i <= n;i++)
    {
        p[i] = -1;
        r[i] = 0;
    }
}

int find_root(int x)
{
    if (p[x] == -1)
        return x;

    int t = p[x];
    p[x] = find_root(p[x]);
    r[x] = (r[x] + r[t]) % 2;
    return p[x];
}

void union_set(int a, int b)
{
    int ra = find_root(a);
    int rb = find_root(b);

    if (ra != rb)
    {
        p[ra] = rb;
        r[ra] = (r[a] + r[b] + 1) % 2;
    }
}

int main()
{
    //freopen("poj1703.txt", "r", stdin);
    int t;
    scanf("%d", &t);
    while (t--)
    {
        int n, m;
        scanf("%d%d", &n, &m);
        make_set(n);
        char c;
        int a, b;
        for (int i = 0;i < m;i++)
        {
            getchar();
            scanf("%c%d%d", &c, &a, &b);
            if (c == 'A')
            {
                if (find_root(a) == find_root(b))    //a,b在一个集合里
                {
                    if (r[a] == r[b])    //a,b为同一帮派
                        puts("In the same gang.");
                    else puts("In different gangs.");    //a,b为不同帮派
                }
                else puts("Not sure yet.");    //a,b不再同一集合里，故不确定
            }
            else union_set(a, b);
        }
    }
    return 0;
}

注意点

1、函数union_set要写成这样：

//正确写法
void union_set(int a, int b)
{
    int ra = find_root(a);
    int rb = find_root(b);

    if (ra != rb)
    {
        p[ra] = rb;
        r[ra] = (r[a] + r[b] + 1) % 2;
    }
}

写成如下形式会MLE：

//错误写法，MLE
void union_set(int a, int b)
{
    int ra = find_root(a);
    int rb = find_root(b);

    p[ra] = rb;
    r[ra] = (r[a] + r[b] + 1) % 2;
}

2、使用scanf输入。

相似题目

1、poj2492