洛谷 P3183 [HAOI2016]食物链

P3183 [HAOI2016]食物链

题目描述

如图所示为某生态系统的食物网示意图，据图回答第1小题现在给你n个物种和m条能量流动关系，求其中的食物链条数。物种的名称为从1到n编号M条能量流动关系形如a1 b1a2 b2a3 b3......am-1 bm-1am bm其中ai bi表示能量从物种ai流向物种bi,注意单独的一种孤立生物不算一条食物链

输入输出格式

输入格式：

 

第一行两个整数n和m,接下来m行每行两个整数ai bi描述m条能量流动关系。（数据保证输入数据符号生物学特点，且不会有重复的能量流动关系出现）1<=N<=100000 0<=m<=200000题目保证答案不会爆 int

 

输出格式：

 

一个整数即食物网中的食物链条数

 

输入输出样例

输入样例#1：

10 16
1 2
1 4
1 10
2 3
2 5
4 3
4 5
4 8
6 5
7 6
7 9
8 5
9 8
10 6
10 7
10 9

输出样例#1：

9

 思路：记忆化搜索。

错因：无。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define MAXN 400000
using namespace std;
long long ans;
int n,m,tot,into[MAXN],out[MAXN];
int f[MAXN],to[MAXN],head[MAXN],net[MAXN];
void add(int u,int v){
    to[++tot]=v;net[tot]=head[u];head[u]=tot;
}
int dfs(int x){
    if(f[x])    return f[x];
    long long bns=0;
    if(into[x]&&!out[x])    bns++;
    for(int i=head[x];i;i=net[i])
        bns+=dfs(to[i]);
    f[x]=bns;
    return bns;
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++){
        int x,y;
        scanf("%d%d",&x,&y);
        into[y]++;out[x]++;
        add(x,y);
    }
    for(int i=1;i<=n;i++)
        if(into[i]==0)
            ans+=dfs(i);
    cout<<ans;
}