【并查集】POJ 1182 食物链

最新推荐文章于 2022-11-17 09:42:34 发布

原创

最新推荐文章于 2022-11-17 09:42:34 发布 · 439 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #并查集

本文详细介绍了POJ 1182食物链问题的解决方法，主要使用并查集数据结构。内容包括问题拆分、两种解题方法的阐述及代码实现解析，探讨了如何处理不同动物类别间的关系，以及如何判断操作是否成功。

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

POJ 1182 : 食物链

由于题目中的N个动物都是A,B,C三类中的其中一种。题目有两种操作方式，【1】”1 X Y”，表示X和Y是同类，这个操作就是并查集中的【并】操作，【2】”2 X Y”，表示X吃Y，这个操作实际上是维护了森林中树之间的关系（类似有向图，此时树类比成一个顶点）。题目求这两个操作不成功的个数，假设为ans。

问题拆分

【1】进行1 X Y表示X和Y是同类之前，首先得判断X是否能吃Y或者Y是否能吃X。如果可以则证明操作不成功，ans累加1，否则将X和Y合并。
【2】2 X Y表示X吃Y时，首先得判断Y是否能吃X或者X和Y是否是同类。如果可以则证明操作不成功，ans累加1，否则建立X和Y的关系。

方法一

通过数组建立分类（A,B,C），首先1~N是A类，（N+1）~ 2 * N是B类，（2 * N + 1）~3 * N是C类。N是动物总数。
上面的判断可以分解为：
【1】X是否能吃Y：X在A类，Y在B类；X在B类，Y是C类；X在C类，Y在A类。
【2】Y是否能吃X：Y在B类，X在A类；Y在C类，X是B类；Y在A类，X在C类。
【3】X和Y是否是同类：X，Y在A类；X，Y在B类； X，Y在C类。

假设输入例子如下

100 8
2 1 2
2 2 3
1 1 4
1 2 5
1 3 6
1 3 9
2 3 2
1 1 2
1 1 3

执行过程演示图如下：
这里写图片描述

代码

#include <iostream>
#include <cstdio>
#include <map>
using namespace std;

const int MAXNUM = 3* (50000 + 10);
int id[MAXNUM];
int Size[MAXNUM];

// 初始化
void make_set(int n){
    for(int i = 1 ; i <= n ; i++){
        id[i] = i;
        Size[i] = 1;
    }
}

// 查找父节点
int Find(int p) {
    while (p != id[p]) {
        // 路径压缩，会破坏掉当前节点的父节点的尺寸信息，因为压缩后，当前节点的父节点已经变了
        id[p] = id[id[p]];
        p = id[p];
    }
    return p;
}

// 合并 p ,q节点
void Union(int p, int q) {
    i