洛谷刷题C++语言 | P4017 最大食物链计数

学习C++从娃娃抓起！记录下洛谷C++学习和备考过程中的题目，记录每一个瞬间。

附上汇总贴：洛谷刷题C++语言 | 汇总_热爱编程的通信人的博客-CSDN博客

---

【题目描述】

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

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

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

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

【输入】

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

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

【输出】

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

【输入样例】

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

【输出样例】

5

【代码详解】

#include <bits/stdc++.h>
#include <vector>
#include <queue>
using namespace std;
typedef pair<int, int> pii;
int n, m, a, b, ans=0;
int in[5005], out[5005], num[5005];
const int mod = 80112002;
vector<pii> s;
vector<int> g[5005];
queue<int> que;
bool cmp(const pii a, const pii b) {  //定义排序函数
    if (a.first == b.first) return a.second<b.second;  //对于起始节点相同的，终止节点按照从小到大排序
    return a.first<b.first;  //起始节点从小到大排序
}
int main()
{
    cin >> n >> m;
    for (int i=1; i<=m; i++) {
        cin >> a >> b;
        out[a]++, in[b]++;
        s.push_back({a, b});
    }
    sort(s.begin(), s.end(), cmp);
    for (int i=0; i<s.size(); i++) {
        g[s[i].first].push_back(s[i].second);
    }
    for (int i=1; i<=n; i++) {
        if (!in[i]) {
            que.push(i);
            num[i]++;
        }
    }
    int temp;
    while (!que.empty()) {
        temp = que.front();
        que.pop();
        for (int i=0; i<g[temp].size(); i++) {
            int v = g[temp][i];
            num[v] = (num[temp] + num[v]) % mod;
            in[v]--;
            if (in[v]==0) {
                que.push(v);
            }
        }
    }
    for (int i=1; i<=n; i++) {
        // cout << num[i] << " ";
        if (out[i]==0) {
            ans = (ans+num[i])%mod;
        }
    }
    cout << ans;
    return 0;
}

【运行结果】

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