力扣笔记（每日随机一题）——最佳买卖股票时机含冷冻期

最新推荐文章于 2024-10-09 20:57:36 发布

ACxz

最新推荐文章于 2024-10-09 20:57:36 发布

阅读量405

点赞数

分类专栏： leetcode每日一题文章标签： leetcode 笔记算法

本文链接：https://blog.csdn.net/pylittlebrat/article/details/131185590

版权

leetcode每日一题专栏收录该内容

3 篇文章 0 订阅

订阅专栏

该问题是一个经典的动态规划问题，给定一个整数数组prices，表示每天的股票价格。目标是在满足卖出后一天不能买入（即存在冷冻期）的条件下，找到最大的交易利润。通过维护三个状态变量：卖出、买入和冷冻期，使用动态规划方法更新每个状态的最大收益。最终返回temp数组的最后一个元素作为最大利润。

摘要由CSDN通过智能技术生成

问题（中等）

给定一个整数数组prices，其中第 prices[i] 表示第 i 天的股票价格。

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

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

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/best-time-to-buy-and-sell-stock-with-cooldown/

示例 1

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

示例 2

输入: prices = [1]
输出: 0

提示：

$1 <= p r i ces . l e n g t h <= 5000$
$0 <= p r i ces [i] <= 1000$

解题

思路

（BP问题太折磨了，思路参考官方解答）
每一天可以有三个状态：买、卖和冷冻期；

temp[i]表示截至第i天，最后一个操作是卖时的最大收益；
buy[i]表示截至第i天，最后一个操作是买时的最大收益；
frozen[i]表示截至第i天，最后一个操作是冷冻期时的最大收益；
递推公式：
temp[i] = max(buy[i-1]+prices[i], temp[i-1]) (第一项表示第i天卖出，第二项表示第i天冷冻)
buy[i] = max(frozen[i-1]-prices[i], buy[i-1]) （第一项表示第i天买进，第二项表示第i天冷冻）
frozen[i] = max(temp[i-1], buy[i-1], frozen[i-1])

代码实现

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
    
        n = len(prices)
        if n == 0:
            return 0     
        temp = [0]*n
        buy = [0]*n
        frozen = [0]*n
        buy[0] = -prices[0]
        for i in range(1,n):
            temp[i] = max(buy[i-1] + prices[i], temp[i-1])
            buy[i] = max(frozen[i-1] - prices[i], buy[i-1])
            frozen[i] = max(temp[i-1], buy[i-1],frozen[i-1])
        return temp[-1]