leetcode怎么算出最佳买卖股票时机含冷冻期

这篇文章主要讲解了“leetcode怎么算出最佳买卖股票时机含冷冻期”,文中的讲解内容简单清晰,易于学习与理解,下面请大家跟着小编的思路慢慢深入,一起来研究和学习“leetcode怎么算出最佳买卖股票时机含冷冻期”吧!

一、题目内容

给定一个整数数组,其中第 i 个元素代表了第 i 天的股票价格 。

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

  • 你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。

  • 卖出股票后,你无法在第二天买入股票 (即冷冻期为 1 天)。

示例:

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

二、解题思路

无脑dp:

第i天卖出 = max(第i-1天买入第i天卖出,第i-1天卖出)

第i天买入 = max(第i-1天冰冻期第i天买入,第i-1天买入)

第i天冰冻期 = max(第i-1天卖出第i天冰冻期,第i-1天冰冻期)

三、代码

class Solution:
    def maxProfit(self, prices: list) -> int:
        if len(prices) == 0:
            return 0
        sell = [0 for _ in range(len(prices))]
        buy = [0 for _ in range(len(prices))]
        freeze = [0 for _ in range(len(prices))]
        buy[0] = -prices[0]
        for i in range(1, len(prices)):
            sell[i] = max(buy[i - 1] + prices[i], sell[i - 1])
            buy[i] = max(freeze[i - 1] - prices[i], buy[i - 1])
            freeze[i] = max(sell[i - 1], freeze[i - 1])
        return sell[-1]


if __name__ == '__main__':
    prices = [1, 2, 3, 0, 2]
    s = Solution()
    ans = s.maxProfit(prices)
    print(ans)

感谢各位的阅读,以上就是“leetcode怎么算出最佳买卖股票时机含冷冻期”的内容了,经过本文的学习后,相信大家对leetcode怎么算出最佳买卖股票时机含冷冻期这一问题有了更深刻的体会,具体使用情况还需要大家实践验证。这里是亿速云,小编将为大家推送更多相关知识点的文章,欢迎关注!