2.数据结构:5.食物链

cpp 复制代码
#include<algorithm>
#include<iostream>
#include<cstring> 

using namespace std;

const int N=50010;

int n,m;
int p[N],d[N];

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

int main(){
    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++;
        }else{
            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]-d[x]+1;
                }
            }
        }
    }
    
    printf("%d\n",ans);
    
    return 0;
}

总共三种动物,维护和根节点之间的距离。对三取模之后把同一个集合里面的动物分为了三类。

相关推荐
别或许1 小时前
1、高数----函数极限与连续(知识总结)
算法
田梓燊1 小时前
code 560
数据结构·算法·哈希算法
笨笨饿1 小时前
29_Z变换在工程中的实际意义
c语言·开发语言·人工智能·单片机·mcu·算法·机器人
kobesdu2 小时前
综合强度信息的激光雷达去拖尾算法解析和源码实现
算法·机器人·ros·slam·激光雷达
weixin_413063212 小时前
记录 MeshFlow-Online-Video-Stabilization 在线稳像
算法·meshflow·实时防抖
会编程的土豆2 小时前
【数据结构与算法】动态规划
数据结构·c++·算法·leetcode·代理模式
炘爚2 小时前
深入解析printf缓冲区与fork进程复制机制
linux·运维·算法
迈巴赫车主3 小时前
蓝桥杯19724食堂
java·数据结构·算法·职场和发展·蓝桥杯
6Hzlia4 小时前
【Hot 100 刷题计划】 LeetCode 78. 子集 | C++ 回溯算法题解
c++·算法·leetcode
Kethy__4 小时前
计算机中级-数据库系统工程师-数据结构-查找算法
数据结构·算法·软考·查找算法·计算机中级