P4017 最大食物链计数

最新推荐文章于 2024-11-03 01:08:04 发布

Erin2333

最新推荐文章于 2024-11-03 01:08:04 发布

阅读量128

点赞数

分类专栏：洛谷文章标签： c语言动态规划

本文链接：https://blog.csdn.net/erin2333/article/details/120078723

版权

洛谷专栏收录该内容

8 篇文章 0 订阅

订阅专栏

本文介绍了两种方法解决食物链计数问题：拓扑排序和DPS+DP（记忆化搜索）。代码示例展示了如何利用C++实现这两种算法，通过维护节点的入度和出度进行遍历或搜索，最终计算食物链的数量，并在过程中使用模运算避免溢出。

摘要由CSDN通过智能技术生成

简单dp题单的最后一道题！

从前有一个人，她想dp，后来写着写着，就写成了topo。

本题这里提供两种做法，拓扑排序和DPS+DP（记忆化搜索），使用vector数组存图，一维变长。

拓扑排序

#include <iostream>
#include <vector>
#include <queue>
using namespace std;

// topo
int n, m, ans;
int in[5005], out[5005], f[5005];
vector<int> food[5005];
queue<int> q;

int main(){
    scanf("%d%d", &n, &m);
    int a, b;
    for(int i = 1; i <= m; i++){
        scanf("%d%d", &a, &b);
        food[a].push_back(b);
        out[a] ++;
        in[b] ++;
    }
    for(int i = 1; i <= n; i ++){
        if(in[i]==0){
            q.push(i);
            f[i] = 1;
        }
    }

    while (!q.empty()){
        int p = q.front();
        q.pop();
        for(int i = 0; i < food[p].size(); i ++){
            int x = food[p][i];
            f[x] += f[p];
            f[x] %= 80112002;
            in[x] --;
            if(in[x] == 0){     
                if(out[x]==0){                 
                    ans += f[x];
                    ans %= 80112002;
                }
                else q.push(x);
            }
        }
    }
    
    printf("%d\n", ans);
    return 0;
}

输入的时候记一下每个点的出度和入度，将入度为0的点入队，f[i]的值等于所有指向它的点的f[pre]之和。f[i]每加上一个f[pre]，把点i的入度-1，如果出度为0说明到头啦，加在ans上，当入度为0时入队。注意不要忘记%80112002！

DPS+DP（记忆化搜索）

#include <iostream>
#include <vector>
#include <queue>
using namespace std;

//dfs+dp
int n, m, ans;
int in[5005], out[5005], dp[5005];
vector<int> food[5005];

int dfs(int x){
    if(out[x]==0) return 1;
    if(dp[x]!=0) return dp[x];
    for(int i = 0; i < food[x].size(); i ++){
        dp[x] += dfs(food[x][i]);
        dp[x] %= 80112002;
    }
    return dp[x];
}

int main(){
    scanf("%d%d", &n, &m);
    int a, b;
    for(int i = 1; i <= m; i++){
        scanf("%d%d", &a, &b);
        food[a].push_back(b);
        out[a] ++;
        in[b] ++;
    }
    for(int i = 1; i <= n; i ++){
        if(in[i] == 0){
            ans += dfs(i);
            ans %= 80112002;
        }
    }
    printf("%d\n", ans);
    return 0;
}