拓扑排序--P4017 最大食物链计数

最大食物链计数

题目背景

你知道食物链吗？Delia 生物考试的时候，数食物链条数的题目全都错了，因为她总是重复数了几条或漏掉了几条。于是她来就来求助你，然而你也不会啊！写一个程序来帮帮她吧。

题目描述

给你一个食物网，你要求出这个食物网中最大食物链的数量。

（这里的“最大食物链”，指的是生物学意义上的食物链，即最左端是不会捕食其他生物的生产者，最右端是不会被其他生物捕食的消费者。）

Delia 非常急，所以你只有 $1$ 秒的时间。

由于这个结果可能过大，你只需要输出总数模上 $80112002$ 的结果。

输入格式

第一行，两个正整数 $n、m$，表示生物种类 $n$ 和吃与被吃的关系数 $m$。

接下来 $m$ 行，每行两个正整数，表示被吃的生物A和吃A的生物B。

输出格式

一行一个整数，为最大食物链数量模上 $80112002$ 的结果。

样例 #1

样例输入 #1

5 7
1 2
1 3
2 3
3 5
2 5
4 5
3 4

样例输出 #1

5

提示

各测试点满足以下约定：

【补充说明】

数据中不会出现环，满足生物学的要求。（感谢 @AKEE ）

wa了十几次，最后发现数组开小了，以后要注意e数组和ne数组大小是边的数量而不是点的数量，，，，，，
存完图之后，把所有入度为0的点入队，每次拿出来一个，把这个点连到的所有点的入度-1， 然后把存的链条数加过去，当有新的入度为0的点，如果这个点出度也为0，那就是食物链顶端，加到答案里，否则还有天敌，继续入队

#include <bits/stdc++.h>
using namespace std;

const int mod = 80112002;

int n, m;
int h[5005], e[500005], ne[500005], idx;
int in[5005], out[5005];
long long cnt[5005];
long long ans;

queue<int> q;

void add(int a, int b)   //添加一条有向边并增加点的入度出度
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++;
    in[b] ++;
    out[a] ++;
}

int main()
{
    std::ios::sync_with_stdio(false);
    cin.tie(0); 

    cin >> n >> m;
    
    memset(h, -1, sizeof h);
    
    for(int i = 1; i <= m; i ++) {
        int a, b;
        cin >> a >> b;
        add(a, b);
    }
    
    for(int i = 1; i <= n; i ++) if(!in[i]) cnt[i] = 1, q.push(i);
    //将入度为0的点入队，cnt数组为1代表有一条食物链
    
    while(!q.empty())
    {
        int t = q.front();
        q.pop();
        for(int i = h[t]; i != -1; i = ne[i]) {
            int p = e[i];
            cnt[p] += cnt[t];
            cnt[p] %= mod;
            if(!--in[p])
            {
                if(!out[p]) ans += cnt[p], ans %= mod;//出度为0，食物链顶端
                else q.push(p);
            }
        }
    }
    
    cout << ans;
    return 0;
}