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

你学会了吗???

相关推荐
努力学算法的蒟蒻几秒前
day26(12.6)——leetcode面试经典150
算法·leetcode·面试
代码游侠5 分钟前
数据结构——哈希表
数据结构·笔记·学习·算法·哈希算法·散列表
ULTRA??9 分钟前
moonbit关于模式匹配中的变量绑定
开发语言·c++·人工智能
FY_201814 分钟前
Stable Baselines3中调度函数转换器get_schedule_fn 函数
开发语言·人工智能·python·算法
CoderYanger23 分钟前
动态规划算法-子数组、子串系列(数组中连续的一段):26.环绕字符串中唯一的子字符串
java·算法·leetcode·动态规划·1024程序员节
Jane-66677740 分钟前
C语言——表达式、语句、函数
c语言·开发语言·算法
Tony_yitao1 小时前
12.华为OD机试 - N个选手比赛前三名、比赛(Java 双机位A卷 100分)
java·算法·华为od·algorithm
小龙报1 小时前
【算法通关指南:数据结构与算法篇】树形结构遍历指南:DFS 递归深搜与 BFS 队列广搜实战解析
c语言·数据结构·c++·算法·链表·深度优先·visual studio
业精于勤的牙1 小时前
浅谈:快递物流与算法的相关性(六)
算法