上接:DP&图论 DAY 1 上午
P3183 [HAOI2016]食物链
题解
◦ 给定n个点m条边的有向无环食物网,求其中有多少条极长食物链。
◦ n<=10^5,m<=2*10^5

>Solution
◦ 拓扑图dp经典题
◦ 设f[u]为以节点u为终点的食物链数量。

◦ 按照拓扑序的顺序转移即可。
处理的时候有些小细节:
1.记录每个点有多少入度:方便topsort
2.有多少出度:出度为0的非单点就是一条食物链的结束,这里可以用一个数组 vis 打标记(万一你记上单点了怎么办QAQ)
3.拓扑序转移:食物链起点的 f[u] 标记为 1 ,不然你以后咋转移???

代码
#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
inline int read()
{
int ans=0;
char last=' ',ch=getchar();
while(ch<'0'||ch>'9') last=ch,ch=getchar();
while(ch>='0'&&ch<='9') ans=ans*10+ch-'0',ch=getchar();
if(last=='-') ans=-ans;
return ans;
}
const int maxn=1e5+10;
const int maxm=2e5+10;
int n,m,ans=0;
int in[maxn],out[maxn],f[maxn];
int head[maxn],cnt=0;
bool vis[maxn];
struct node
{
int u,v,nxt;
}edge[maxm];
void addedge(int u,int v)
{
edge[++cnt].u =u;edge[cnt].v =v;
edge[cnt].nxt =head[u];head[u]=cnt;
}
void topsort()
{
queue<int>q;
for(int u=1;u<=n;u++)
if(!in[u])
q.push(u),f[u]=1;
while(!q.empty() )
{
int u=q.front() ;
q.pop() ;
for(int i=head[u];i;i=edge[i].nxt )
{
int v=edge[i].v ;
f[v]+=f[u];
in[v]--;
if(!in[v]) q.push(v);
}
}
}
int main()
{
n=read();m=read();
for(int i=1,u,v;i<=m;i++)
{
u=read();v=read();
addedge(u,v);
in[v]++;
out[u]++;
}
for(int i=1;i<=n;i++){ //打标记
if(!out[i]&&in[i])
vis[i]=1;
}
topsort();
for(int i=1;i<=n;i++){
// printf("f[%d]=%d ",i,f[i]);
if(vis[i])
ans+=f[i];
}
printf("%d",ans);
return 0;
}