bzoj4562 [Haoi2016]食物链

题目

输入符合生物学特点233，所以这就应该是一个有向无环图，我们尝试在上面做dp。

显然，f[i]=sigma(f[u]),u连向x，按照拓扑序来。
注意统计答案时，只记录出度为0的点。
当然，一个点不能算食物“链”233

#include<bits/stdc++.h>
#define N 100000
using namespace std;
int n,m,x,y,ans;
int f[N+1],rd[N+1],cd[N+1];
vector <int> edge[N+1];
queue <int> Q;
inline char nc()
{
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int read()
{
    int x=0,b=1;
    char c=nc();
    for(;!(c<='9'&&c>='0');c=nc())if(c=='-')b=-1;
    for(;c<='9'&&c>='0';c=nc())x=x*10+c-'0';
    return x*b;
} 
inline void add(int x,int y)
{
    edge[x].push_back(y);
}
int main()
{
    //freopen("in.txt","r",stdin);
    n=read(),m=read();
    for(int i=1;i<=m;i++)
    {
        x=read(),y=read();
        cd[x]++,rd[y]++;
        add(x,y);
    }
    for(int i=1;i<=n;i++)
        if(!rd[i])
            if(cd[i])f[i]=1,Q.push(i);
    while(!Q.empty())
    {
        int x=Q.front();Q.pop();
        for(int i=0;i<edge[x].size();i++)
        {
            int u=edge[x][i];
            f[u]+=f[x];rd[u]--;
            if(!rd[u])Q.push(u);
        }
    }
    for(int i=1;i<=n;i++)if(!cd[i])ans+=f[i];
    cout<<ans;
    return 0;
}