[POJ 1182]食物链[并查集]

题目链接:[POJ 1182]食物链[并查集]

题意分析:

动物间可能存在一种食物链x吃y,y吃z,z吃x。下面给出一些语句,判断有多少句是假话。

解题思路:

设置三个并查集,分别代表物种,同一并查集中的动物种类相同。

不同种类间的合并代表着捕食关系。例如 x, y。因为x可以为任意一个种类中的动物,所以在合并它们捕食关系时,我们要把所有的可能情况都合并起来。

什么时候说谎话呢?

1.超出了编号范围。

2.说是同种,但是相互间捕食了。

3.说是捕食,但确是同种,或者捕食关系反过来了。即:说a吃b,但是却已经b吃a了。

个人感受:

神奇,神奇,神奇,重要的事要说三遍。嘛,也是我第三次做这个题了orz

具体代码如下:

#include<cstdio>

const int MAXN = 5e4 + 111;

int p[3 * MAXN], ran[3 * MAXN];

void init(int up)
{
    for (int i = 0; i <= up; ++i) p[i] = i, ran[i] = 0;
}

int find(int x)
{
    return p[x] == x ? x : p[x] = find(p[x]);
}

bool same(int x, int y)
{
    return find(x) == find(y);
}

void unite(int x, int y)
{
    x = find(x), y = find(y);
    if (x == y) return;
    if (ran[x] > ran[y]) p[y] = x;
    else
    {
        p[x] = y;
        if (ran[x] == ran[y]) ++ran[x];
    }
}

int main()
{
    int n, k, ans = 0;
    scanf("%d%d", &n, &k);
    init(3 * n);
    int op, a, b;
    for (int i = 0; i < k; ++i)
    {
        scanf("%d%d%d", &op, &a, &b);
        if (a < 1 || a > n || b < 1 || b > n) //超出编号范围
        {
            ++ans;
            continue;
        }
        if (op == 1)
        {
            if (same(a, b + n) || same(b, a + n)) //互相存在捕食关系
                ++ans;
            else
            {
                unite(a, b);
                unite(a + n, b + n);
                unite(a + 2*n, b + 2*n);
            }
        }
        else
        {
            if (same(a, b) || same(b, a + n)) //同种;反捕食
                ++ans;
            else
            {
                unite(a, b + n);
                unite(a + n, b + 2*n);
                unite(a + 2*n, b);
            }
        }
    }
    printf("%d\n", ans);
    return 0;
}


你可能感兴趣的:(并查集)