BZOJ 4562 [Haoi2016]食物链 - 计数DP+拓扑排序

好像弱到连拓扑排序都忘记了是什么了。。。

大概就是通过路径删入度，删到入度为零进队。而这道题正好利用了这样的思想，对于每一种生物都是先需要统计其上的所有方案数，然后将方案数传给子节点，最后输出汇点方案数即可。

主要需要特判孤立点（即入度出度均为零的点），这种点不能算作一条链，初始值赋为零。

#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<queue>
#include<algorithm>
 
using namespace std;
 
const int maxn=100005;
 
struct edge
{
    int to,next;
}e[maxn<<1];
 
int n,m,ans,cnt;
int in[maxn],out[maxn],head[maxn];
int dp[maxn];
 
void insert(int a,int b)
{
    e[++cnt].to=b;e[cnt].next=head[a];head[a]=cnt;
}
int main()
{
    scanf("%d%d",&n,&m);
    while(m--)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        insert(a,b);
        in[b]++,out[a]++;
    }
    queue<int>q;
    for(int i=1;i<=n;i++)
        if(!in[i])
    {
        q.push(i);
        if(out[i])dp[i]=1;//special judge
    }
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        for(int i=head[u];i;i=e[i].next)
        {
            int v=e[i].to;
            dp[v]+=dp[u];
            in[v]--;
            if(!in[v])q.push(v);
        }
    }
    for(int i=1;i<=n;i++)
        if(!out[i])ans+=dp[i];
    printf("%d",ans);
    return 0;
}