【bzoj4562】[Haoi2016]食物链 记忆化搜索

题目链接： 点击打开链接

题意：中文题意，就不解释了。

题解：当时想了好几种的方法，但是都超时了，原因在于没用想到用记忆化搜索。用记忆化搜索可以大大的节省时间复杂度。

对于每个点来说，只需要查询该点是否是第一次访问，若是第一次访问就继续向下访问，第二次访问就直接返回上次一的条数值。若当前节点为出度为0的点，那么就该点的权值++；

比较简单。看一下代码的

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

const int SZ = 400010;
const int INF = 1e9+10;

int in[SZ],out[SZ];
long long dp[SZ];
int head[SZ],nxt[SZ],to[SZ];
void build(int f,int t){
	static int tot = 1;
	to[++tot] = t;
	nxt[tot] = head[f];
	head[f] = tot;
}
long long dfs(int u){
	if(dp[u]) return dp[u];
	long long ans =0 ;
	if(out[u] == 0 && in[u]) ans ++;
	for(int i = head[u] ; i ; i = nxt[i])
		ans += dfs(to[i]);
	return dp[u] = ans ;
}
int main(){
	int n,m;
	scanf("%d%d",&n,&m);
	for(int i = 1; i <= m ; i ++){
		int a,b;
		scanf("%d%d",&a,&b);
		build(a,b);
		in[b]++;
		out[a] ++;
	}
	long long ans = 0;
	for(int i = 1; i <= n ; i ++){
		if(!in[i])
			ans += dfs(i);
	}
	cout << ans << endl;
}