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

股票类问题(动态规划)

---

题目详情

给定一个整数数组prices，其中第 prices[i] 表示第i天的股票价格 。
设计一个算法计算出最大利润。在满足以下约束条件下，你可以尽可能地完成更多的交易（多次买卖一支股票）:
卖出股票后，你无法在第二天买入股票 (即冷冻期为 1 天)。
注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。​

---

示例1：

输入: prices = [1,2,3,0,2]
输出: 3 
解释: 对应的交易状态为: [买入, 卖出, 冷冻期, 买入, 卖出]

示例2:

输入: prices = [1]
输出: 0

思路:
对于含有冷冻期的股票问题,我们可以添加两个状态-----持股状态s1和未持股状态s2
相应的我们可以画出状态图:

我们对于每天的股票,更新它的buy,sell,s1,s2
这里需要注意的是更新buy的时候,与其他三个不同
因为对于buy的这一天,肯定不是昨天卖出的股票,即不能取sell[i-1]状态
细节详细见代码:

我的代码：

class Solution 
{
public:
    int maxProfit(vector<int>& prices) 
    {
        int n = prices.size();
        if (n == 0) return 0;

        vector<int> buy(n), sell(n), s1(n), s2(n);
        s1[0] = buy[0] = -prices[0]; //第一天买入.持股的状态初始化,此时利润-prices[0]
        sell[0] = s2[0] = 0;         //对于第一天的状态,我们只能买入,所以sell[0]和s2[0]无意义,初始化为0
        //遍历每天的股票
        for (int i = 1; i < n; ++i)
        {
        //既然要买入,那么肯定手中没有股票,这里不能取
        //s2[i-1]和sell[i-1]的max,因为有一天冷冻期,这个细节在这里体现   
            buy[i] = s2[i-1] - prices[i];
        //持股状态,max{昨天买入的,或者昨天已经有股票(更早买入的)}   
            s1[i] = max(buy[i-1], s1[i-1]);
        //既然要卖出,那么手中一定有股票
        //             max{昨天买的     昨天已经有股票(更早买入的)}  +卖出的价格    
            sell[i] = max(buy[i-1], s1[i-1]) + prices[i];
        //未持股状态  max{昨天就未持股(早卖了) ,  昨天卖了}   
            s2[i] = max(s2[i-1], sell[i-1]);
        }
        //返回最后一天卖出和最后一天未持股两个状态的较大值
        return max(sell[n-1], s2[n-1]); //注意下标从0开始所以是n-1
    }
};

涉及知识点:

1.动态规划(dp)