目录

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

相似问题:

分析

  • 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