洛谷P3183/HAOI2016 食物链

@Pelom  November 3, 2019

题意

给出$n$个物种和$m$条关系,求食物链条数

数据范围:$1 \le n \le 100000,0 \le m \le 200000$

题解

记忆化搜索
食物链从入度为$0$的点开始,到出度为$0$的点结束
注意单独的一种孤立生物不算一条食物链

代码:

#include<iostream>
#include<cstdio>
using namespace std;
const int N=100000+10;
int n,m,u,v;
int cnt,ans;
int head[N],in[N],out[N],dp[N];
struct edge{
    int nxt,to;
} e[N<<1];
inline void add(int u,int v){
    e[++cnt]=(edge){head[u],v};
    head[u]=cnt;
}
int dfs(int u){
    if(dp[u])
        return dp[u];
    if(!out[u])
        return dp[u]=1;
    for(int i=head[u];i;i=e[i].nxt)
        dp[u]+=dfs(e[i].to);
    return dp[u];
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++){
        scanf("%d%d",&u,&v);
        add(u,v);
        out[u]++;
        in[v]++;
    }
    for(int i=1;i<=n;i++)
        if(!in[i] && out[i])
            ans+=dfs(i);
    printf("%d",ans);
    return 0;
}

添加新评论