洛谷 P4017 Topological sorting 模板

例题洛谷 P4017：https://www.luogu.com.cn/problem/P4017

求最大食物链的数量，题中已给出最大食物链的定义，可以转化定义一条链的起点就是入度为0的点，出度为0的点可以作为终点。注意食物网抽象出的有向图可能有多个。

思路：只需一遍拓朴排序，动态累计记录一下答案即可，最终累加输出到终点的最大食物链数量即可。

模板:

bool toposort() {
    q = new queue();
    for (i = 0; i < n; i++)
        if (in_deg[i] == 0) q.push(i);
    ans = new vector();
    while (!q.empty()) {
        u = q.pop();
        ans.push_back(u);
        for each edge(u, v) {
            if (--in_deg[v] == 0) q.push(v);
        }
    }
    if (ans.size() == n) {
        for (i = 0; i < n; i++)
            std::cout << ans[i] << std::endl;
        return true;
    } else {
        return false;
    }
}

AC代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 5e3+5;
const ll mod = 80112002;
struct node{
    vector<int> v;
    int in,out;
}g[maxn];
int n,m;
ll ans[maxn];
void add(int u,int v){
    g[v].v.push_back(u);
    g[v].out++,g[u].in++;   
}
void toposort(){
    queue<int> q;
    for(int i = 1;i<=n;i++) if(!g[i].in ) q.push(i),ans[i] = 1;
    while(!q.empty()){
        int cur = q.front();
        q.pop();
        for(int i = 0;i<g[cur].v.size();i++){
            int u = g[cur].v[i];
            ans[u] = (ans[u] + ans[cur])%mod;
            if(--g[u].in == 0) q.push(u);  
        }
    }
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i = 1;i<=m;i++){
        int u,v;
        scanf("%d%d",&u,&v);
        add(u,v);
    }
    toposort();
    ll res = 0;
    for(int i = 1;i<=n;i++){
        if(!g[i].out ) res = (res+ans[i])%mod;
    }
    cout<<res;
    return 0;
}