luogu 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

输出格式

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

---

#include<cstdio>
#include<iostream>
using namespace std;
const int N=1e5+10,M=4e5+10;
int Next[M],head[N],go[M],tot;
int in[N],ou[N],f[N];
inline void add(int u,int v){
	Next[++tot]=head[u];head[u]=tot;go[tot]=v;
}
int dfs(int x){
	if(ou[x]==0)return 1;
	int ans=0;
	
	for(int i=head[x];i;i=Next[i]){
		int y=go[i];
		if(f[y])ans+=f[y];
		else {
			f[y]=dfs(y);
			ans+=f[y];
		}
	}
	
	return ans;
}
int main(){
	int n,m;
	cin>>n>>m;
	for(int i=1,a,b;i<=m;i++){
		scanf("%d%d",&a,&b);
		add(a,b);
		in[b]++;ou[a]++;
	}
	int ans=0;
	for(int i=1;i<=n;i++){
		if(in[i]==0&&ou[i]!=0)
		ans+=dfs(i);
	}
	cout<<ans<<endl;
}