leetcode 309. 买卖股票的最佳时机含冷冻期

2023.8.22

        本题是买卖股票系列 + 冷冻期。 由于引入了冷冻期，并且这个冷冻期是在卖出股票才会出现，因此我dp数组设置了四种状态：

• 状态一：持有股票。
• 状态二：不持有股票： 之前就卖了，所以今天不处于冷冻期。
• 状态三：不持有股票：今天卖。
• 状态四：不持有股票：昨天卖的，所以今天处于冷冻期。

        下面分别分析四种状态：

1. 持有股票：①早就有了：dp[i-1][0]  ②今天刚买：昨天可能是状态二或者状态四，不可能是状态三，因为今天可以买股票，所以不处于冷冻期，所以昨天不可能卖股票。即：dp[i-1][1]-prices[i] 或者 dp[i-1][3]-prices[i] 。 
2. 不持有股票（之前就卖了，今天不处于冷冻期）：①昨天的之前就已经卖了，即dp[i-1][1]。 ②昨天的昨天卖的，即昨天处于冷冻期，今天不处于冷冻期，即dp[i-1][3]。
3. 不持有股票（今天卖）： 没什么好说的，昨天持有股票的钱+今天股票的股价，即dp[i-1][0] + prices[i]
4. 不持有股票（昨天卖，今天处于冷冻期）：已经知道是昨天卖的了，即dp[i-1][2]。

        最后结果取不持有股票的三种情况的最大值即可。 代码如下：

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        vector<vector<int>> dp(prices.size(),vector<int>(4));
        //初始化
        dp[0][0] = -prices[0];
        dp[0][1] = dp[0][2] = dp[0][3] = 0;
        //状态一：持有股票。
        //状态二：不持有股票： 之前就卖了，所以今天不处于冷冻期。
        //状态三：不持有股票：今天卖。
        //状态四：不持有股票：昨天卖的，所以今天处于冷冻期。
        for(int i=1; i<prices.size(); i++)
        {
            dp[i][0] = max(dp[i-1][0] , max(dp[i-1][1]-prices[i] , dp[i-1][3]-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[prices.size()-1][1],max(dp[prices.size()-1][2],dp[prices.size()-1][3]));
    }
};

        附上我的草稿以供参考：