【BZOJ】4562 [Haoi2016]食物链

$Description$

现在给你$n$个物种和$m$条能量流动关系，求其中的食物链条数。
物种的名称为从$1$到$n$编号
$M$条能量流动关系形如
$a_1$ $b_1$
$a_2$ $b_2$
$a_3$ $b_3$
$......$
$a_{m-1}$ $b_{m-1}$
$a_m$ $b_m$
其中$a_i$ $b_i$表示能量从物种$a_i$流向物种$b_i$,注意单独的一种孤立生物不算一条食物链

$Input$

第一行两个整数$n$和$m$,接下来$m$行每行两个整数$a_i$ $b_i$描述$m$条能量流动关系。
（数据保证输入数据符号生物学特点，且不会有重复的能量流动关系出现）
$1<=N<=100000 0<=m<=200000$
题目保证答案不会爆 $int$

$Output$

一个整数即食物网中的食物链条数

$Solution$

据说这题题意坑了很多现场的选手……
这题做法应该很多……我写的是拓扑序DP
（即一个点所有前置点都已经处理过了，我才处理这个点）

#include<stdio.h>
#define N 100001
#define M 200001

int s[N],in[N],out[N],d[N],n,m,u,v,ans;
bool b[N];

struct edge{int v,n;}e[M];

void dfs(const int &k)
{
    for (int i=s[k];i;i=e[i].n)
    {
        d[e[i].v]+=d[k];
        in[e[i].v]--;
        if (in[e[i].v]==0) dfs(e[i].v);
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=m;i++)
    {
        scanf("%d%d",&u,&v);
        out[u]++;
        in[e[i].v=v]++;
        e[i].n=s[u];
        s[u]=i;
    }
    for (int i=1;i<=n;i++) if (in[i]==0 && s[i]) b[i]=1;
    for (int i=1;i<=n;i++) if (in[i]==0 && b[i]) d[i]=1,dfs(i);
    for (int i=1;i<=n;i++) if (out[i]==0) ans+=d[i];
    printf("%d",ans);
}