poj 1182 食物链 并查集 题解《挑战程序设计竞赛》

地址 http://poj.org/problem?id=1182

 

 题解

可以考虑使用并查集解决 但是并不是简单的记录是否同一组的这般使用

每个动物都有三个并查集 自己 天敌 捕食 并查集

那么在获得一条信息后 我们先判断真伪

x不能吃x 自己

x y不能超过数目类型

当xy是同一类的时候  x不会出现在y的天敌和捕食并查集中(其中已经包含 y不会出现在x的天敌和捕食并查集中)

确认为真后 合并更新 x y 的同类并查集   天敌并查集和 不是并查集

 

当x吃y的信息， 则 x不会出现在y的同类和捕食并查集中(已经包含y不会出现在x同类且y不会出现在x的天敌并查集中)

确认为真后 合并更新 x和y的天敌并查集 合并更新 x的天敌与y的捕食并查集 合并更新x的捕食与y的并查集

 

代码

#include <iostream>
#include <algorithm>
#include <vector>


using namespace std;

const int MAX_N = 900010;

int par[MAX_N];        //父节点
int rankk[MAX_N];    //树的高度

//初始化n个元素
void init(int n)
{
    for (int i = 0; i < n; i++) {
        par[i] = i;
        rankk[i] = 0;
    }
}

//查询树的根
int find(int x) {
    if (par[x] == x) {
        return x;
    }
    else {
        return par[x] = find(par[x]);
    }
}

//合并x和y所属于的集合
void unite(int x, int y) {
    x = find(x);
    y = find(y);
    if (x == y) return;

    if (rankk[x] < rankk[y]) {
        par[x] = y;
    }
    else {
        par[y] = x;
        if (rankk[x] == rankk[y]) rankk[x]++;
    }
}

//判断x和y是否属于同一个集合
bool same(int x, int y) {
    return find(x) == find(y);
}
//=============================================================
int N, K;
int T[MAX_N], X[MAX_N], Y[MAX_N];

void solve()
{
    //初始化并查集
    //元素x ,x+N,x+2*N分别代表x-A x-B x-C
    init(N * 3);

    int ans = 0;
    for (int i = 0; i < K; i++) {
        int t = T[i];
        int x = X[i] - 1, y = Y[i] - 1;

        //不正确的编号
        if (x < 0 || N <= x || y < 0 || N <= y) {
            ans++;
            continue;
        }

        if (t == 1) {
            //xy属于同一类
            if (same(x, y + N) || same(x, y + 2 * N)) {
                ans++;
            }
            else {
                unite(x, y);
                unite(x + N, y + N);
                unite(x + 2 * N, y + 2 * N);
            }
        }
        else {
            //x吃y
            if (same(x, y) || same(x, y + 2 * N)) {
                ans++;
            }
            else {
                unite(x, y + N);
                unite(x + N, y + 2 * N);
                unite(x + 2 * N, y);
            }
        }
    }

    printf("%d\n", ans);
}


int main()
{
    cin >> N >> K;

    for (int i = 0; i < K; i++) {
        cin >> T[i] >> X[i] >> Y[i];
    }

    solve();
    return 0;
}