并查集:240. 食物链

cpp 复制代码
#include<bits/stdc++.h>

using namespace std;

const int N=50000+10;

int p[N],d[N];

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

int main()
{
    int n,m;
    cin>>n>>m;
    
    for(int i=1;i<=n;i++)   p[i]=i;
    
    int ans=0;
    while(m--)
    {
        int t,x,y;
        cin>>t>>x>>y;
        
        if(x>n||y>n)
        {
            ans++;
            continue;
        }
        
        int px=find(x),py=find(y);
        if(t==1)
        {
            if(px==py&&(d[x]-d[y])%3)   ans++;
            else if(px!=py)
            {
                p[px]=py;
                d[px]=d[y]-d[x];
            }
        }
        else 
        {
            if(px==py&&(d[x]-d[y]-1)%3) ans++;
            else if(px!=py)
            {
                p[px]=py;
                d[px]=d[y]+1-d[x];
            }
        }
    }
    
    cout<<ans<<endl;
    
    return 0;
}

使用并查集维护一些额外信息,每一个节点到根节点的距离对3取模,表示该节点和根节点的关系,假设取模之后的结果是0,表示和根节点是同类,假设结果是1,表示可以吃根节点,结果是2,表示可以被根节点吃,只有三种情况,同类,吃根节点,被根节点吃,同类之间不可以互相吃

函数调用的时候不可以多次调用,否则会发生错误

要知道两个生物之间的关系,可以根据他们和根节点之间的关系来判断,比如说,一个到根节点的距离是1(x),一个到根节点的距离是2(y),那么 yx

情况有多种,没有穷举完所有情况不要使用 else

假设单次修改会改变我们后面要使用的变量,那么我们用一个中间变量(或者叫临时变量)把将被修改的变量存下来,例如

cpp 复制代码
		int u=find(p[x]);
        d[x]+=d[p[x]];
        p[x]=u;

长度的变化可以画示意图来进行理解

p存的是父节点,d存的是到根节点的距离,只保证到根节点的距离是有意义的,这样子可以保证时间复杂度是O(n)

相关推荐
机器学习之心19 小时前
NGO-VMD北方苍鹰算法优化变分模态分解+皮尔逊系数+小波阈值降噪+信号重构,MATLAB代码
算法·matlab·重构·信号重构·ngo-vmd·皮尔逊系数·小波阈值降噪
橘颂TA19 小时前
【剑斩OFFER】算法的暴力美学——山脉数组的蜂顶索引
算法·leetcode·职场和发展·c/c++
速易达网络19 小时前
C语言常见推理题
java·c语言·算法
freedom_1024_19 小时前
LRU缓存淘汰算法详解与C++实现
c++·算法·缓存
博语小屋20 小时前
力扣11.盛水最多的容器(medium)
算法·leetcode·职场和发展
Swift社区20 小时前
LeetCode 423 - 从英文中重建数字
算法·leetcode·职场和发展
点云SLAM20 小时前
算法与数据结构之二叉树(Binary Tree)
数据结构·算法·二叉树·深度优先·广度优先·宽度优先
小龙报20 小时前
《算法通关指南:算法基础篇 --- 一维前缀和 — 1. 【模板】一维前缀和,2.最大子段和》
c语言·数据结构·c++·算法·职场和发展·创业创新·visual studio
树在风中摇曳21 小时前
LeetCode 1658 | 将 x 减到 0 的最小操作数(C语言滑动窗口解法)
c语言·算法·leetcode
不夜牛仔21 小时前
算法笔记17 - 贪心算法介绍与思路 | 路灯摆放问题 | 活动安排问题 | 最低字典序拼接 | 金条分割问题 | 项目投资问题
笔记·算法·贪心算法