【Leetcode刷题】：Python：309. 最佳买卖股票时机含冷冻期

Jack_Kuo

已于 2022-04-07 15:32:38 修改

阅读量211

点赞数

CC 4.0 BY-SA版权

分类专栏： Python刷Leetcode 文章标签： leetcode python 算法动态规划数据结构与算法

于 2022-01-20 12:30:01 首次发布

本文链接：https://blog.csdn.net/weixin_37251044/article/details/122598955

Python刷Leetcode 专栏收录该内容

135 篇文章

订阅专栏

本文介绍了一种使用动态规划解决股票买卖问题的方法，特别是考虑到冷冻期这一限制条件。通过定义三种状态来跟踪不同情况下的最大收益，并逐步迭代计算最终的最优解。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目

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

代码：动态规划：O(n)

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

        n = len(prices)
        f = [[0, 0, 0] for _ in range(n)]
        # 第一天持有，只有可能是买入
        f[0][0] = -prices[0]
        # f[i][0]: 手上持有股票的最大收益
        # f[i][1]: 手上不持有股票，本日卖出，累计最大收益
        # f[i][2]: 手上不持有股票，本日不卖出，累计最大收益
        for i in range(1, n):
            # 今日持有：1.昨日持有；2.昨日不持有，今日买入
            f[i][0] = max(f[i-1][0], f[i-1][2]-prices[i])
            # 今日不持有，因为今日卖出：1.昨日持有，今日卖出
            f[i][1] = f[i-1][0] + prices[i]
            # 今日不持有，因为本身没有：1.昨日卖出了；2.昨日也没有
            f[i][2] = max(f[i-1][1], f[i-1][2])
        
        return max(f[n-1][1], f[n-1][2])