力扣-动态规划-309 买卖股票的最佳时机含冷冻期

思路

1. dp数组定义：
2. 递推公式：

   dp[i][0] = max(dp[i - 1][0], max(dp[i - 1][3] - prices[i], 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];

3. dp数组初始化：dp[0][0] -= prices[0]
4. 时间复杂度：     

代码

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        vector<vector<int>>dp(prices.size(), vector<int>(4, 0));
        // 0 持有 1 不持有 2 今天卖出   3 冷冻状态
        dp[0][0] -= prices[0];

        for(int i = 1; i < prices.size(); i++){
            dp[i][0] = max(dp[i - 1][0], max(dp[i - 1][3] - prices[i], 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[prices.size()-1][1], max(dp[prices.size()-1][3], dp[prices.size()-1][2]));

    }
};