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

题目描述

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

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

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

示例 1:

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

解法

参考：Leetcode 121. 买卖股票的最佳时机

有如下递推关系式：

这里的 表示前一天不持有股票，并且再前一天也不持有股票，即当天不处于冷却期，可以买入股票。

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        if not prices:
            return 0
        lastI0,i0,i1=0,0,-prices[0]
        for i in range(1,len(prices)):
            lastI0,i0,i1=i0,max(i0,i1+prices[i]),max(i1,lastI0-prices[i])
        return i0