力扣 -- 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];
    }
};

你学会了吗???

相关推荐
wearegogog1231 小时前
基于块匹配的MATLAB视频去抖动算法
算法·matlab·音视频
十重幻想1 小时前
PTA6-1 使用函数求最大公约数(C)
c语言·数据结构·算法
青岛少儿编程-王老师2 小时前
CCF编程能力等级认证GESP—C++5级—20250927
java·数据结构·c++
脑子慢且灵2 小时前
C语言与Java语言编译过程及文件类型
java·c语言·开发语言·汇编·编辑器
蒙奇D索大2 小时前
【C语言加油站】C语言文件操作详解:从“流”的概念到文件的打开与关闭
c语言·开发语言·笔记·学习·改行学it
大千AI助手2 小时前
蛙跳积分法:分子动力学模拟中的高效数值积分技术
算法·积分·数值积分·蛙跳积分法·牛顿力学系统·verlet积分算法
zycoder.3 小时前
力扣面试经典150题day3第五题(lc69),第六题(lc189)
算法·leetcode·面试
你的冰西瓜3 小时前
C++动态规划入门指南——助力CSP竞赛夺冠
c++·动态规划
小此方3 小时前
C语言自定义变量类型结构体理论:从初见到精通(上)
c语言·开发语言
西阳未落4 小时前
LeetCode——双指针
c++·算法