洛谷 P3183 BZOJ 4562 [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

解题思路

　　记忆化搜索即可。BZOJ难得有一道水题

源代码

#include<stdio.h>
int n,m;
int num[100010]={0},ans;
struct Edge{
    int next,to;
}e[200010];
int cnt=1,head[100010]={0},ru[100010]={0};
void add(int u,int v)
{
    e[cnt]={head[u],v};
    head[u]=cnt++;
    ru[v]++;
}

int dfs(int u)
{
    if(num[u]) return num[u];
    if(!head[u]) return 1;
    int num_u=0;
    for(int i=head[u];i;i=e[i].next)
    {
        int v=e[i].to;
        num_u+=dfs(v);
    }
    return num[u]=num_u;
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=0,u,v;i<m;i++)
    {
        scanf("%d%d",&u,&v);
        add(u,v);
    }
    for(int i=1;i<=n;i++)
        if(ru[i]==0&&head[i]) ans+=dfs(i);
    printf("%d
",ans);
    return 0;
}