买卖股票的系列问题五!
这回的限制是含有冷冻期。
题目
leetcode 309
代码与思路
确定状态
明确状态关系,四种
冷冻期,卖出股票后不能再买一次
dp[i][j],第i天状态为j,所剩的最多现金为dp[i][j]。
- 状态0:买入股票状态(今天买入股票,或者是之前就买入了股票然后没有操作)
- 卖出股票状态,这里就有两种卖出股票状态
- 状态1:两天前就卖出了股票,度过了冷冻期,一直没操作,今天保持卖出股票状态
- 状态2:今天卖出了股票
- 状态3:今天为冷冻期状态,但冷冻期状态不可持续,只有一天!
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];
//前一天是卖出状态
初始化
如果是持有股票状态(状态0)那么:dp[0][0] = -prices[0],买入股票所剩现金为负数。
保持卖出股票状态(状态1),第0天没有卖出dp[0][1]初始化为0就行,
今天卖出了股票(状态2),同样dp[0][2]初始化为0,因为最少收益就是0,绝不会是负数。
同理dp[0][3]也初始为0。
遍历答案
//买卖股票的最佳时机(含冷冻期)
class Solution {
public:
int maxProfit(vector<int>& prices) {
int n = prices.size();
if (n == 0) return 0;
vector<vector<int>> dp(n, vector<int>(4, 0));
dp[0][0] -= prices[0]; // 持股票
for (int i = 1; i < n; i++) {
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];
}
//返回几个状态中的最大值
return max(dp[n - 1][3],max(dp[n - 1][1], dp[n - 1][2]));
}
};