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;
}

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

相关推荐
山烛11 分钟前
决策树学习全解析:从理论到实战
人工智能·python·学习·算法·决策树·机器学习
修己xj1 小时前
探索设计模式的宝库:Java-Design-Patterns
算法
鲨鱼辣椒_TUT1 小时前
MySQL连接算法和小表驱动大表的原理
mysql·算法·adb
设计师小聂!1 小时前
力扣热题100------21.合并两个有序链表
算法·leetcode·链表
এ᭄画画的北北2 小时前
力扣-1.两数之和
数据结构·算法·leetcode
shenghaide_jiahu2 小时前
数学建模——递归和动态规划
算法·数学建模·动态规划
黄昏晓x3 小时前
数据结构----排序
java·数据结构·排序算法
凯子坚持 c3 小时前
动态规划专题:详解二维费用背包问题——以“一和零”与“盈利计划”为例
算法·动态规划
黑色的山岗在沉睡3 小时前
P1948 [USACO08JAN] Telephone Lines S
数据结构·c++·算法·图论