[BZOJ4562] [Haoi2016]食物链

Description
如图所示为某生态系统的食物网示意图，据图回答第1小题
现在给你n个物种和m条能量流动关系，求其中的食物链条数。
物种的名称为从1到n编号
M条能量流动关系形如
a1 b1
a2 b2
a3 b3
……
am-1 bm-1
am bm
其中ai bi表示能量从物种ai流向物种bi,注意单独的一种孤立生物不算一条食物链

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

输出：

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

样例输入：

10 16
1 2
1 4
1 10
2 3
2 5
4 3
4 5
4 8
6 5
7 6
7 9
8 5
9 8
10 6
10 7
10 9

样例输出：

9

题解：（退役选手的BZOJ生涯）对于每个点计算他们的入度和出度。我们设入度为0的点为起始点。出度为0的点为终止点。那么只需要以入度为0的点做一遍拓扑排序就可以统计答案了。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<queue>
#include<vector>
#define LiangJiaJun main
#define pa pair<int,int>
#define INF 1999122700ll
using namespace std;
int n,m,h[100004],ne=1;
struct edge{
    int to , nt;
}e[450004];
vector<int>ed;
int ru[100004],cu[100004];
int ans[100004],res;
queue<int>q;
void insert(int u,int v){e[++ne].to = v; e[ne].nt = h[u]; h[u] = ne;}
int LiangJiaJun(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++){
        int u,v;
        scanf("%d%d",&u,&v);
        insert(u,v);
        ++ru[v];
        ++cu[u];
    }
    for(int i=1;i<=n;i++)if(cu[i]==0&&ru[i]==0)res--;
    for(int i=1;i<=n;i++){
        if(cu[i]==0)ed.push_back(i);
        if(ru[i]==0){
            q.push(i);
            ans[i]=1;
        }
    }
    while(!q.empty()){
        int x = q.front(); q.pop();
        for(int i=h[x];i;i=e[i].nt){
            ans[e[i].to]+=ans[x];
            ru[e[i].to]--;
            if(ru[e[i].to]==0)q.push(e[i].to);
        }
    }
    for(int i=0;i<ed.size();i++)res += ans[ed[i]];
    cout<<res<<endl;
    return 0;
}