原题链接: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]));
}
};