★★★[并查集变形] AcWing 240. 食物链 题解

news/2024/7/15 19:34:04 标签: 图论, c++

输入样例:

100 7
1 101 1 
2 1 2
2 2 3 
2 3 3 
1 1 3 
2 3 1 
1 5 5

输出样例:

3

 思路:

我们可以将食物链的环状捕食关系用并查集的形式表现出来, 印射成某节点距根节点的距离取余3的值(0 /  1  /  2 ), 差值为1的即为捕食与被捕食的关系, 如 2 - > 1 - > 0 ->2表示的是2吃1, 1吃0(0可以吃2 )

重点:

1. 在压缩并查集时, 同时需要注意节点到根的距离d

2. D=1时, 

如果X, Y(此篇统一写作 a, b)均已纳入该生态系统(并查集合), 即find(a)==find(b) , 再判断他们与根的距离是否满足 da%3 - db%3 = 0 (化简为 (da-db)%3 == 0 )

如果X, Y中有未纳入的,即find(a)!=find(b), 接下来我们需要将他们的根合并, 但这里与以往并查集合并不同的是, 不只要把他们的祖先合并, 还得维护与根的距离使得捕食关系(即取余3)不发生变化,

我们不妨让pa(a的根) 等于pb, 此时要使a, b各自距根的距离取余3后

相等,对照下图, 即(da+?-db)%3 == 0, 我们可以让 ? (即d[pa]) 满足 d[pa] = db-da

D=2时, 与上面的思路大体相似

如果X, Y(此篇统一写作 a, b)均已纳入该生态系统(并查集合), 即find(a)==find(b) , 再判断他们与根的距离是否满足 da%3 - db%3 = 1(即a吃b) (化简为 (da-db-1)%3 == 0 )

如果X, Y中有未纳入的,即find(a)!=find(b), 接下来我们需要将他们的根合并, 但这里与以往并查集合并不同的是, 不只要把他们的祖先合并, 还得维护与根的距离使得捕食关系(即取余3)不发生变化,

我们不妨让pa(a的根) 等于pb, 此时要使a, b各自距根的距离取余3后

相等,对照下图, 即(da+?-db)%3 == 1 (化简为(da+?-db-1)%3 == 0), 我们可以让 ? (即d[pa]) 满足 d[pa] = db-da+1

#include <iostream>
using namespace std;

const int N = 50010;
int p[N], d[N]; //可食用关系,食物链等级(距根节点的距离取余后分为0,1,2三个等级)

int find(int x)
{
    if(p[x]!=x){
        int t = find(p[x]);
        d[x] += d[p[x]];
        p[x] = t;
    }
    return p[x];
}

int main()
{
    int n,m;
    cin >> n >> m;
    
    for (int i = 1; i <= n; i ++ ) p[i] = i;
    
    int op, a, b;
    int res = 0;
    while (m -- )
    {
        cin >> op >> a >> b;
        
        if(a>n || b>n ) res ++;
        else
        {
            int pa = find(a), pb = find(b);
            if(op == 1){
                if(pa==pb && (d[a]-d[b])%3)res ++;
                else if(pa!=pb){
                    p[pa] = pb;
                    d[pa] = d[b] - d[a];
                }
            }
            else{
                if(pa==pb && (d[a]-d[b]-1)%3) res ++;
                else if(pa!=pb){
                    p[pa] = pb;
                    d[pa] = d[b]-d[a]+1;
                }
            }
        }
    }
    cout << res;
    return 0;
}


http://www.niftyadmin.cn/n/816657.html

相关文章

数据结构笔记4(排序 上)

冒泡排序 基本原理&#xff1a;以从小到大排序为例&#xff0c;每次从上到下比较两个相邻的值&#xff0c;如果不是从小到大的顺序则互换这两个值的位置。符合顺序则不变。这样从第一个到最后一个筛选一遍&#xff0c;第一趟排序结束&#xff0c;此时最大的数已经放在了最后面…

Manjaro Linux (KDE桌面)安装以及使用中的一些问题

昨天晚上一顿折腾终于装好了Manjaro&#xff0c;觉得有必要写篇博客记录一下以免重装的时候没有参考。 提醒&#xff1a;开始的某些操作是错误操作&#xff0c;千万别一步一步跟着操作&#xff0c;阅读一下后文&#xff01;&#xff01; 坎坷的安装过程 遇到的第一个问题在于启…

[D-OJ 练习] (取余优化) 判断一个字符串不区分大小写是否回文

输入描述 输入一个字符串&#xff0c;以#号结束&#xff0c;不包含空格&#xff0c;占一行 输出描述 判断字符串不区分大小写是否回文&#xff0c;是输出YES&#xff0c;否则输出NO。 输入样例 MadaDAm# 输出样例 YES 写的时候在判断大小写上有了一些想法,可以省去一些…

数据结构笔记5(排序 下)

快速排序&#xff08;quick sort&#xff09; 算法概述 分而治之 例如&#xff1a;从下图中随便选一个数字作为主元pivot&#xff0c;图中选了65.接下来以65为中枢&#xff0c;将原来的集合分为两大块&#xff0c;左边这一块全部小于65&#xff0c;右边全部大于65.加下来递归…

2021辽宁省大学生程序设计竞赛 C D E F G I L

WAWA大哭, 弱鸡七题摘银准备退役... 感谢队友方大师和棋兄的鼎力相助&#x1f44d; C传染病统计https://ac.nowcoder.com/acm/contest/22352/C 题目描述 阿强来到大街上&#xff0c;街上有 N 个人&#xff0c;编号为 1 ∼N 。简单起见&#xff0c;我们把每个人都看成一条线上…

数据结构笔记6 散列查找

散列表 目前已知的查找方法&#xff1a; 但是上述这几种方法对于长字符串的动态查找效果都不是很好。看下面的例子&#xff1a; 可以看到二分法在这个问题中效果不是很好。那么如何快速搜索到需要的关键词&#xff1f;如果关键词不方便比较怎么办&#xff1f; 查找的本质&am…

leecode刷题笔记-数组

数组题注意事项&#xff1a; 1. 切记while循环的循环条件一定要判断遍历长度是否越界且要先判断该条件&#xff0c;否则就会报错&#xff0c;例如&#xff1a; while j<len(nums2) and nums2[j]<i 这里如果改为 while nums2[j]<i and j<len(nums2) 就可能会报错&a…

[邻接矩阵形式]无向图的建立与深度,广度遍历

目录 MGraph类 构造函数 深度优先遍历 广度优先遍历 MGraph类 const int N 10; int visit[N]; // 顶点是否被访问 template<typename DataType> class MGraph//无向图 { public:MGraph(DataType a[], int n, int e);~MGraph(){}void DF(int x); //深度优先遍历void…