leetcode [309] 最佳买卖股票时机含冷冻期

---

Contact me:

Blog -> https://cugtyt.github.io/blog/index
Email -> cugtyt@qq.com
GitHub -> Cugtyt@GitHub

---

给定一个整数数组，其中第 i 个元素代表了第 i 天的股票价格 。​

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

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

示例:

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

思路来自题解:

0 不持有：
    原来持有，现在卖出
    原来卖出，等待

1 持有： 
    原来持有，继续持有
    原来卖出，买入

price   1   2   3   0   2
0       0   1   2   1   1
1       -1  -1  -1  1   3

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        if len(prices) < 2:
            return 0
        dp = [[0] * 2 for _ in range(len(prices) + 1)]
        dp[1][0] = 0
        dp[1][1] = -prices[0]
        for i in range(2, len(prices) + 1):
            dp[i][0] = max(prices[i - 1] + dp[i - 1][1], dp[i - 1][0])
            dp[i][1] = max(dp[i - 1][1], dp[i - 2][0] - prices[i - 1])
        return max(dp[-1])