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

原题链接：309. 最佳买卖股票时机含冷冻期

 

solution:

状态机，本题包含三种状态，如下图所示

 

状态表示：

dp[i][0]： 下标i天时，处于冷冻期时时所获得的最大利润。

dp[i][1]：下标i天时，持有股票时获得的最大利润

dp[i][2]： 下标i天时，卖完股票时所获得的最大利润。

状态计算：

dp[i][0] = max(dp[i - 1][0],dp[i - 1][2])

dp[i][1] = max(dp[i - 1][0] - p[i],dp[i - 1][1])

dp[i][2] = dp[i - 1][1] + p[i],

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size();
        vector<vector<int>> dp (n, vector<int> (3, INT_MIN));
        dp[0][1] = -prices[0];
        dp[0][0] = 0;
        for(int i = 1;i < n;i++) {
            dp[i][0] = max(dp[i - 1][0],dp[i - 1][2]);
            dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] - prices[i]);
            dp[i][2] = dp[i - 1][1] + prices[i];
        }
        return max(dp[n - 1][0],max(dp[n - 1][1],dp[n - 1][2]));
    }
};