AcWing 2716. 食物链
原题链接
中等
作者:
我木jj我怕谁
,
2022-02-17 22:43:18
,
所有人可见
,
阅读 271
算法1
(记忆化搜索!)
参考文献
C++ 代码
#include<iostream>
#include<cstring>
using namespace std;
const int N = 3e6+9;
int ne[N],e[N],h[N],idx;
int n,m;
int f[N];
int in[N],out[N];
void add(int a,int b)
{
e[idx] = b,ne[idx] = h[a],h[a] = idx++;
}
int dfs(int u)
{
if(out[u] == 0) return 1;
int sum = 0;
if(f[u]) return f[u];
for(int i=h[u]; i!=-1; i=ne[i])
{
int j = e[i];
sum = sum + dfs(j);
}
f[u] = sum;
return sum;
}
int main()
{
int ans = 0;
memset(h,-1,sizeof h);
scanf("%d%d",&n,&m);
for(int i=1; i<=m; i++)
{
int a,b;
scanf("%d%d",&a,&b);
add(a,b);
in[b]++;
out[a]++;
}
for(int i=1; i<=n; i++)
{
if(!in[i] && out[i]) //入度为0,但是出度不为0;
{
ans += dfs(i);
}
}
printf("%d\n",ans);
return 0;
}
%%%