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

你学会了吗???

相关推荐
XiYang-DING6 分钟前
【LeetCode】 225.用队列实现栈
算法·leetcode·职场和发展
ybzj.29 分钟前
2025年第十六届蓝桥杯省赛C/C++大学B组 个人题解
c++
花月C29 分钟前
线性动态规划(Linear DP)
算法·动态规划·代理模式
派大星~课堂33 分钟前
【力扣-148. 排序链表】Python笔记
python·leetcode·链表
量子炒饭大师1 小时前
【C++ 11】Cyber骇客 最后的一片净土 ——【C++11的 简单介绍 + 发展历史】历史唯物主义者带你理顺C++发展的由来
c++·dubbo·c++11
hetao17338371 小时前
2025-03-24~04-06 hetao1733837 的刷题记录
c++·算法
小白菜又菜1 小时前
Leetcode 657. Robot Return to Origin
python·leetcode·职场和发展
_深海凉_1 小时前
LeetCode热题100-环形链表
算法·leetcode·链表
原来是猿1 小时前
Linux进程信号详解(三):信号保存
开发语言·c++·算法
2401_892070981 小时前
算法与数据结构精讲:最大子段和(暴力 / 优化 / 分治)+ 线段树从入门到实战
c++·算法·线段树·最大子段和