「HAOI2016」食物链 - 记忆化搜索

题目链接:传送门

Description

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

Input

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

Output

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

Solution

这是一道记搜的模板题。我们从入度为0的点开始搜,到出度为0的点结束,设f[x]表示x的答案,然后记搜即可

Code

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
#include<bits/stdc++.h>
#define begin Begin
#define next Next
#define in In
#define out Out
using namespace std;
const int maxn=200000+100;
int n,m,e;
int begin[maxn],to[maxn*2],next[maxn*2];
int in[maxn],out[maxn];
inline void add(int x,int y){
to[++e]=y;
next[e]=begin[x];
begin[x]=e;
}
int f[maxn];
inline int dfs(int x){
if(f[x])return f[x];
int ans=0;
if(!out[x] && in[x])ans++;
for(int i=begin[x];i;i=next[i])
ans+=dfs(to[i]);
f[x]=ans;
return ans;
}
int main(){
#ifndef ONLINE_JUDGE
freopen("a.in","r",stdin);
freopen("a.out","w",stdout);
#endif
scanf("%d%d",&n,&m);
int x,y;
for(int i=1;i<=m;i++){
scanf("%d%d",&x,&y);
add(x,y);
in[y]++;
out[x]++;
}
int ans=0;
for(int i=1;i<=n;i++)
if(!in[i])
ans+=dfs(i);
printf("%d\n",ans);
return 0;
}
感谢你的赞赏!