HYSBZ - 4562——食物链 (记忆化搜索)

如图所示为某生态系统的食物网示意图，据图回答第1小题

现在给你n个物种和m条能量流动关系，求其中的食物链条数。

物种的名称为从1到n编号

M条能量流动关系形如

a1 b1

a2 b2

a3 b3

......

am-1 bm-1

am bm

其中ai bi表示能量从物种ai流向物种bi,注意单独的一种孤立生物不算一条食物链

Input

第一行两个整数n和m,接下来m行每行两个整数ai bi描述m条能量流动关系。

（数据保证输入数据符号生物学特点，且不会有重复的能量流动关系出现）

1<=N<=100000 0<=m<=200000

题目保证答案不会爆 int

Output

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

Sample Input 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

Sample Output 9

思路:记忆化搜索。

#include<algorithm>
#include<string.h>
#include<stdio.h>
#define M 100010
using namespace std;
int first[M],k;
int bigen[M],enden[M],v[M];
struct node
{
    int y,next;
}a[2*M];
void add(int x,int y)
{
    a[k].y=y;
    a[k].next=first[x];
    first[x]=k++;
    bigen[x]++;
    enden[y]++;
}
int dfs(int x)
{
    int ans=0;
    if(!bigen[x]) return 1;
    if(v[x]) return v[x];//返回记忆结果
    int t=first[x];
    while(t>=0)
    {
        ans+=dfs(a[t].y);
        t=a[t].next;
    }
    v[x]=ans;//记忆化搜索
    return ans;
}
int main()
{
    int n,m,x,y,ans;
    while(~scanf("%d%d",&n,&m))
    {
        k=ans=0;
        memset(first,-1,sizeof(first));
        memset(bigen,0,sizeof(bigen));
        memset(enden,0,sizeof(enden));
        memset(v,0,sizeof(v));
        for(int i=0;i<m;i++)
        {
            scanf("%d%d",&x,&y);
            add(x,y);
        }
        for(int i=1;i<=n;i++)
        {
            if(!enden[i]&&bigen[i])
            {
                ans+=dfs(i);
            }
        }
        printf("%d\n",ans);
    }
}