poj-1182-食物链(并查集）

最新推荐文章于 2020-07-29 16:30:42 发布

齐豪

最新推荐文章于 2020-07-29 16:30:42 发布

阅读量437

点赞数

CC 4.0 BY-SA版权

分类专栏：数据结构

本文链接：https://blog.youkuaiyun.com/qq_33096883/article/details/77621659

数据结构专栏收录该内容

12 篇文章

订阅专栏

本文介绍如何使用并查集解决POJ 1182问题，通过创建三倍数量的节点来表示每种动物可能所属的三种类型，并通过特定规则进行合并以检查冲突。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目地址

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

解题思路

对于每只动物i创造三个元素i-A, i-B, i-C, 并用这3*N个元素建立并查集。
- i-x表示“i属于种类x”
- 并查集里的每一组内所有元素代表的情况都同时发生或不发生
例如，如果i-A和j-B在同一个组里，那么就表示如果i属于种类A那么j一定属于种类B，如果j属于种类B那么i一定属于种类A。所以：
- x,y同一类—–合并x-A和y-A, x-B和y-B,x-C-和y-C
- x吃y————合并x-A和y-B, x-B和y-C,x-C-和y-A
在合并之前，判断是否会有矛盾。

Code

#include <iostream>
#include <stack>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <stdio.h>
#include <map>
#include <queue>
#include <vector>
#include <algorithm>
#define INF 0x3fffffff
#define N 50005
#define K 100005

using namespace std;

int n, k;
int fa[3*N];
int rk[3*N];

void init() {
    for (int i = 0; i < 3*n; i++) {
        fa[i] = i;
        rk[i] = 0;
    }
}

int _find(int x) {
    if (x == fa[x]) return x;
    else return fa[x] = _find(fa[x]);
}

void _union(int x, int y) {
    x = _find(x);
    y = _find(y);
    if (x == y) return;
    if (rk[x] < rk[y]) {
        fa[x] = y;
    } else {
        fa[y] = x;
        if (rk[x] == rk[y]) rk[x]++;
    }
}

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

int main()
{
#ifndef ONLINE_JUDGE
    freopen("in.txt","r",stdin);
#endif
    scanf("%d%d", &n, &k);
    init();
    int ans = 0;
    for (int i = 0; i < k; i++) {
        int d, x, y;
        scanf("%d%d%d", &d, &x, &y);
        x--; y--;
        if (!(0 <= x && x < n && 0 <= y && y < n)) {
            ans++;
            continue;
        }
        if (d == 1) {
            if (same(x, y+n) || same(x, y+2*n)) {
                ans++;
            } else {
                _union(x, y);
                _union(x+n, y+n);
                _union(x+2*n, y+2*n);
            }
        } else {
            if (same(x, y) || same(x, y+2*n)) {
                ans++;
            } else {
                _union(x, y+n);
                _union(x+n, y+2*n);
                _union(x+2*n, y);
            }
        }
    }
    cout << ans << endl;
    return 0;
}