https://leetcode.cn/problems/best-time-to-buy-and-sell-stock-with-cooldown/
给定一个整数数组prices,其中第 prices[i] 表示第 i 天的股票价格 。设计一个算法计算出最大利润。在满足以下约束条件下,你可以尽可能地完成更多的交易(多次买卖一支股票):卖出股票后,你无法在第二天买入股票 (即冷冻期为 1 天)。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
示例 1:
输入: prices = [1,2,3,0,2]
输出: 3
解释: 对应的交易状态为: [买入, 卖出, 冷冻期, 买入, 卖出]
class Solution:
def maxProfit(self, prices: List[int]) -> int:
# 难度:☆☆☆
# 标签:动态规划
# 不限制买卖次数,冻结期1天,本题是122的变形
# 有冻结期,意味着本次买的上一次卖的,而是上上次卖的
# 那么就需要多一个步骤,多记录上一次卖的状态
buy, sell_pre, sell_now = -float('inf'), 0, 0
for p in prices:
buy = max(buy, sell_pre - p) # 这次买接的是上上次卖的
sell_pre = sell_now # 更新sell_pre
sell_now = max(sell_now, buy + p)
return sell_now