HX-Note

image

导读 ^ _ ^

买卖股票的系列问题五!
这回的限制是含有冷冻期。

题目

leetcode 309
image.png

代码与思路

确定状态

明确状态关系,四种
冷冻期,卖出股票后不能再买一次
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]));
    }
};

#谢谢你的观看!

^ _ ^

相关文章: