洛谷P4017 最大食物链计数 题解 DAG最长路计数(动态规划)

题目链接:https://www.luogu.com.cn/problem/P4017

解题思路:

定义 \(f[i]\) 为到点 \(i\) 的方案数,则:

\[f[i] = \sum_{j \to i}{f[j]} \]

其中 \(j \to i\) 表示存在一条从 \(j\)\(i\) 的边。

实现代码如下:

#include <bits/stdc++.h>
using namespace std;
const long long MOD = 80112002LL;
const int maxn = 5050;
int n, m, in[maxn], out[maxn];
long long f[maxn], ans;
bool vis[maxn];
vector<int> g[maxn];
long long dfs(int u) {
    if (!out[u]) return 1;
    if (vis[u]) return f[u];
    vis[u] = true;
    int sz = g[u].size();
    for (int i = 0; i < sz; i ++) {
        int v = g[u][i];
        f[u] = (f[u] + dfs(v)) % MOD;
    }
    return f[u];
}
int main() {
    cin >> n >> m;
    while (m --) {
        int a, b;
        cin >> a >> b;
        in[a] ++;
        out[b] ++;
        g[b].push_back(a);
    }
    for (int i = 1; i <= n; i ++)
        if (!in[i])
            ans = (ans + dfs(i)) % MOD;
    cout << ans << endl;

    return 0;
}
posted @ 2020-02-17 11:27  quanjun  阅读(214)  评论(0编辑  收藏  举报