代码随想录算法训练营day42|动态规划part9

今天的第一题和第三题都比较简单,只需在前面的题目的代码上略加修改即可。第二题是有冷冻期的股票问题,这题的状态分析比前面的题目难不少。

卡哥的做法是分为四个状态,分别是持有股票,保持不持有股票的状态,卖出股票的状态以及冷冻期4个状态。根据定义可列出状态转移方程如下:

复制代码
dp[i][0] = max(dp[i - 1][0], max(dp[i - 1][3], dp[i - 1][1]) - prices[i]);
dp[i][1] = max(dp[i - 1][1], dp[i - 1][3]);
dp[i][2] = dp[i - 1][0] + prices[i];
dp[i][3] = dp[i - 1][2];

这题非常巧妙的地方在于相较于前面两题,这题将保持不持有股票的状态和卖出股票的状态分开,这样有利于进行冷冻期的操作。

接下来进行初始化,dp[0][0]的意思是在第0天持有股票时的现金,自然是0.接着dp[0][1],这里不能直接根据字面意思推断,是因为这里dp[0][1]本身没有意义,所以要根据状态转移方程推导出dp[0][1]=0,因为如果初始化为其它数值答案都不正确。

我自己用的方法是用flag标记冷冻期,然后分类讨论:

复制代码
 vector<vector<int>>dp(prices.size(),vector<int>(2,0));
        int flag=0;//标记冷冻期状态
        dp[0][0]=0;
        dp[0][1]=-prices[0];
        //0表示卖出,1表示买入;
        for(int i=1;i<prices.size();i++){
            
            if(flag==0||i<2){
                dp[i][1]=max(dp[i-1][1],dp[i-1][0]-prices[i]);
            }
               else if(flag==1&&i>=2){//防止数组越界
               // dp[i][0]=max(dp[i-2][0],dp[i-2][1]+prices[i]);
                dp[i][1]=max(dp[i-2][1],dp[i-2][0]-prices[i]);
                flag=0;//在冷冻期后的一次处理后更新冷冻期状态
           }
           dp[i][0]=max(dp[i-1][0],dp[i-1][1]+prices[i]);
            if(dp[i][0]==dp[i-1][1]+prices[i]) {
                    flag=1;//更新冷冻期状态
                }
            
        }

但是卡哥的思路还是更为直观的

相关推荐
小O的算法实验室18 小时前
2026年ASOC,基于深度强化学习的无人机三维复杂环境分层自适应导航规划方法,深度解析+性能实测
算法·无人机·论文复现·智能算法·智能算法改进
郭涤生20 小时前
STL vector 扩容机制与自定义内存分配器设计分析
c++·算法
༾冬瓜大侠༿20 小时前
vector
c语言·开发语言·数据结构·c++·算法
Ricky111zzz20 小时前
leetcode学python记录1
python·算法·leetcode·职场和发展
汀、人工智能20 小时前
[特殊字符] 第58课:两个正序数组的中位数
数据结构·算法·数据库架构··数据流·两个正序数组的中位数
liu****20 小时前
第16届省赛蓝桥杯大赛C/C++大学B组(京津冀)
开发语言·数据结构·c++·算法·蓝桥杯
汀、人工智能20 小时前
[特殊字符] 第79课:分割等和子集
数据结构·算法·数据库架构·位运算·哈希表·分割等和子集
汀、人工智能20 小时前
[特殊字符] 第74课:完全平方数
数据结构·算法·数据库架构·图论·bfs·完全平方数
CoderCodingNo20 小时前
【GESP】C++四、五级练习题 luogu-P1177 【模板】排序
数据结构·c++·算法
Proxy_ZZ020 小时前
从零实现LDPC比特翻转译码器:C语言实战与底层逻辑解析
c语言·算法