309. 最佳买卖股票时机含冷冻期 - 力扣(LeetCode)
股票系列:
LeetCode第 121 题:买股票的最佳时机(C++)_zj-CSDN博客
LeetCode第 122 题:买股票的最佳时机II(C++)_zj-CSDN博客
LeetCode第 123 题:买股票的最佳时机III(C++)_zj-CSDN博客
LeetCode第 188 题:买股票的最佳时机IV(C++)_zj-CSDN博客
LeetCode第 309 题:最佳买卖股票时机含冷冻期(C++)_zj-CSDN博客
LeetCode第 714 题:买卖股票的最佳时机含手续费(C++)_zj-CSDN博客
相比之前的题目加了一个限制,交易次数依然是无限次。
LeetCode第 122 题:买股票的最佳时机II(C++)_zj-CSDN博客
那么只需要在之前的状态转移上做一点修改,也就是买入的时候,状态需要从前两天转移过来,其他的都是一样的:
class Solution {
public:
int maxProfit(vector<int>& prices) {
int n = prices.size();
if(n < 2) return 0;
//第一维表示日期,第二维表示是否持有股票(1:有,0:无),数组值表示收益
vector<vector<int>> dp(n, vector<int>(2, -1e7));
dp[0][0] = 0; //第一天没有股票,没买没卖,收益为0
dp[0][1] = 0 - prices[0]; //第一天有股票,说明买入了,收益为负(花掉一笔钱)
dp[1][0] = max(dp[0][0], dp[0][1]+prices[1]);//昨天就没有今天rest,或者昨天有今天进行了卖出
dp[1][1] = max(dp[0][1], dp[0][0]-prices[1]);//昨天就有今天rest,或者昨天没有今天进行买入
for(int i = 2; i < n; ++i){
dp[i][0] = max(dp[i-1][0], dp[i-1][1] + prices[i]);//前一天就没有今天rest,或者前一天有今天卖出
dp[i][1] = max(dp[i-1][1], dp[i-2][0] - prices[i]);//前一天就有今天rest,或者前两天卖出、今天买入
}
//return max(dp[n-1][0], dp[n-1][1]);要想收益最大,最后一天肯定不会进行买入,所以dp[n-1][0]肯定更大
return dp[n-1][0];
}
};
可以进行状态压缩,上述dp只涉及三个变量:
class Solution {
public:
int maxProfit(vector<int>& prices) {
int n = prices.size();
if(n < 2) return 0;
int a = 0, b = -prices[0];//第一天两种状态(未持有、持有)
int c = max(a, b+prices[1]);//第二天未持有
int d = max(b, a-prices[1]);//第二天持有
for(int i = 2; i < n; ++i){
int tmp1 = c;
c = max(c, d+prices[i]);
d = max(d, a-prices[i]);
a = tmp1;
}
return c;
}
};
看了题解之后,发现其实状态的定义也可以再增加一个,也就是持有、未持有、冷冻期,不过总体思路都是差不多的。