LeetCode 热题 HOT 100 第七十一天 309. 最佳买卖股票时机含冷冻期 中等题 用python3求解

题目地址

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

注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。

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

示例 2:
输入: prices = [1]
输出: 0

提示：
1 <= prices.length <= 5000
0 <= prices[i] <= 1000

解法：参考指路

将购买股票，分成三个状态：空仓（可购买）、空仓（不可购买）、持有，三种状态之间的转移如下图：

利用三个变量表示上述三个状态：empty, empty_freeze, hold， 分别表示三种状态在当天的最大收益，将上图转移关系公式化（代码化）：

empty = max(empty, empty_freeze)
empty_freeze = hold + price
hold = max(empty - price, hold)

（举例：empty_freeze = hold + price表示状态为empty_freeze的时候，手里头获得的利益就是原来持有的钱+卖出股票后的钱(price就是当天股票的成交价)）

代码实现：

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        if not prices: 
            return 0

        empty, empty_freeze, hold = 0, 0, -prices[0]

        for i in range(1, len(prices)):
            # 利用元组形式赋值避免了建立临时变量
            empty, empty_freeze, hold = max(empty, empty_freeze), hold + prices[i], max(empty - prices[i], hold)

            # 上面一行代码等价于以下四行代码，需要临时变量temp存储上个交易日结束的empty数值
            #temp = empty
            #empty = max(temp, empty_freeze)
            #empty_freeze = hold + prices[i]
            #hold = max(temp - prices[i], hold)
            
        return max(empty, empty_freeze)

注意：
利用元组形式赋值避免了建立临时变量

 empty, empty_freeze, hold = max(empty, empty_freeze), hold + prices[i], max(empty - prices[i], hold)

上面一行代码等价于以下四行代码，需要临时变量temp存储上个交易日结束的empty数值

temp = empty
empty = max(temp, empty_freeze)
empty_freeze = hold + prices[i]
hold = max(temp - prices[i], hold)