2023.8.22
本题是买卖股票系列 + 冷冻期。 由于引入了冷冻期,并且这个冷冻期是在卖出股票才会出现,因此我dp数组设置了四种状态:
- 状态一:持有股票。
- 状态二:不持有股票: 之前就卖了,所以今天不处于冷冻期。
- 状态三:不持有股票:今天卖。
- 状态四:不持有股票:昨天卖的,所以今天处于冷冻期。
下面分别分析四种状态:
- 持有股票:①早就有了:dp[i-1][0] ②今天刚买:昨天可能是状态二或者状态四,不可能是状态三,因为今天可以买股票,所以不处于冷冻期,所以昨天不可能卖股票。即:dp[i-1][1]-prices[i] 或者 dp[i-1][3]-prices[i] 。
- 不持有股票(之前就卖了,今天不处于冷冻期):①昨天的之前就已经卖了,即dp[i-1][1]。 ②昨天的昨天卖的,即昨天处于冷冻期,今天不处于冷冻期,即dp[i-1][3]。
- 不持有股票(今天卖): 没什么好说的,昨天持有股票的钱+今天股票的股价,即dp[i-1][0] + prices[i]
- 不持有股票(昨天卖,今天处于冷冻期):已经知道是昨天卖的了,即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]));}
};
附上我的草稿以供参考: