力扣_动态规划5—买卖股票的最佳时机含冷冻期

题目

给定一个整数数组prices，其中第 prices[i] 表示第 i 天的股票价格 。​

设计一个算法计算出最大利润。在满足以下约束条件下，你可以尽可能地完成更多的交易（多次买卖一支股票）:

• 卖出股票后，你无法在第二天买入股票 (即冷冻期为 1 天)。
  注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。

方法

• 每天可能处于一下状态：
  • 今天或者今天之前 买了一支股票 $d{p}_1$
  • 今天卖了手里的股票 $d{p}_2$
  • 昨天或者昨天之前 卖了一只股票 $d{p}_3$
• 状态转移
  • $d{p}_1=max(d{p}_1,d{p}_3-prices[i])$ # 第一项表示今天之前买入，第二项表示今天买入
  • $d{p}_2=d{p}_1+prices[i]$ # 今天卖了
  • $d{p}_3=max(d{p}_2,d{p}_3)$ # 第一项表示昨天卖了，第二项表示昨天之前卖了

代码

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size();
        int dp1 = -prices[0]; // 今天或者今天之前 买了一只股票
        int dp2 = 0; // 今天卖了一只股票
        int dp3 = 0; // 昨天或者昨天之前 卖了一张股票
        for(int i = 1; i < n; i++){
            int dp1_temp = max(dp1, dp3-prices[i]);
            int dp2_temp = dp1+prices[i];
            int dp3_temp = max(dp2, dp3);
            dp1 = dp1_temp;
            dp2 = dp2_temp;
            dp3 = dp3_temp;
        }
        return max(dp2, dp3);
    }
};