0309：买卖股票的最佳时机含冷冻期（★）

力扣第 309 题

题目

给定一个整数数组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

相似问题：

• 0121：买卖股票的最佳时机
• 0122：买卖股票的最佳时机 II

分析

• 0122 升级版，dp[i][0] 的递推式有变化：

$$dp[i][0] = max(dp[i-1][0], dp[i-2][1]-prices[i-1])$$

• 可以优化为 3 个变量

解答

1
2
3
4
5
6

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        b0,a,b = 0,-inf,0
        for x in prices:
            b0,a,b = b,max(a,b0-x),max(b,a+x)
        return b

42 ms