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

你学会了吗???

相关推荐
再__努力1点几秒前
【77】积分图像:快速计算矩形区域和核心逻辑
开发语言·图像处理·人工智能·python·算法·计算机视觉
唯唯qwe-15 分钟前
Day22: 贪心算法 | 区间问题,左/右端点排序
算法·贪心算法
Hcoco_me27 分钟前
LLM(Large Language Model)系统学习路线清单
人工智能·算法·自然语言处理·数据挖掘·聚类
java修仙传33 分钟前
力扣hot100:寻找旋转排序数组中的最小值
算法·leetcode·职场和发展
胖咕噜的稞达鸭37 分钟前
算法日记专题:位运算II( 只出现一次的数字I II III 面试题:消失的两个数字 比特位计数)
c++·算法·动态规划
fantasy_arch38 分钟前
pd_process.c 文件源码分析
c语言·数据库·视频编解码·av1
茉莉玫瑰花茶39 分钟前
ProtoBuf - 3
服务器·c++·protobuf
txzrxz1 小时前
图的存储
算法·深度优先·图论
Algebraaaaa1 小时前
为什么线程阻塞要用.join而不是.wait
java·c++·python
Knight_AL1 小时前
深入解析 JVM 垃圾回收算法:经典 vs 新型 GC 算法
jvm·算法