力扣 -- 879. 盈利计划(二维费用的背包问题)

解题步骤:

参考代码:

未优化的代码:

cpp 复制代码
class Solution {
public:
    int profitableSchemes(int n, int minProfit, vector<int>& group, vector<int>& profit) {

        //计划数
        int len=group.size();
        
        //每一维都多开一行空间
        vector<vector<vector<int>>> dp(len+1,vector<vector<int>>(n+1,vector<int>(minProfit+1)));
        
        //初始化
        for(int j=0;j<=n;j++)
        {
            dp[0][j][0]=1;
        }

        //填表
        for(int i=1;i<=len;i++)
        {
            //记得从0开始
            for(int j=0;j<=n;j++)
            {
                //记得从0开始
                for(int k=0;k<=minProfit;k++)
                {
                    //状态转移方程
                    dp[i][j][k]+=dp[i-1][j][k];
                    if(j>=group[i-1])
                    {
                        //max(0,k-profit[i-1])非常重要
                        dp[i][j][k]=(dp[i][j][k]+dp[i-1][j-group[i-1]][max(0,k-profit[i-1])])%(1000000000+7);
                    }
                }
            }
        }

        return dp[len][n][minProfit];
    }
};

优化后的代码:

cpp 复制代码
class Solution {
public:
    int profitableSchemes(int n, int minProfit, vector<int>& group, vector<int>& profit) {
        //计划数
        int len=group.size();
        
        vector<vector<int>> dp(n+1,vector<int>(minProfit+1));
        
        //初始化
        for(int j=0;j<=n;j++)
        {
            dp[j][0]=1;
        }

        //填表
        for(int i=1;i<=len;i++)
        {
            //记得从大到小遍历
            for(int j=n;j>=group[i-1];j--)
            {
                //记得从大到小遍历
                for(int k=minProfit;k>=0;k--)
                {
                    //状态转移方程
                    dp[j][k]=(dp[j][k]+dp[j-group[i-1]][max(0,k-profit[i-1])])%(1000000000+7);
                }
            }
        }
        //返回值
        return dp[n][minProfit];
    }
};

你学会了吗???

相关推荐
晓蛋2 天前
c语言指的是什么意思
c语言·编译器·编程开发·集成开发环境·程序实例
倒头就睡的小比特2 天前
算法竞赛C++常用的STL
c++·算法
weilx12342 天前
C++笔记-文件IO-<fcntl.h>
c++
小羊没烦恼!2 天前
初探性能优化——2个月到4小时的性能提升
java·开发语言·windows·算法·c#
傲世仙尊2 天前
目录即文件-Ext文件系统收尾篇
linux·c语言
牵猫散步的鱼儿2 天前
重载、重写(覆盖)、重定义区别
c语言
猎头南楼2 天前
知识社区推荐系统实践:新用户冷启动与长短期兴趣建模的挑战 资深推荐算法工程师
人工智能·深度学习·算法·机器学习
phltxy2 天前
C 语言指针:从内存地址到灵活的数据访问
c语言
Smileyqp沛沛2 天前
前端?C++ ?较大差异基础罗列
c++·基础·前端转c++
m0_547486662 天前
《数据结构教程》全套 PPT课件2026
数据结构