最佳买卖股票时机含冷冻期Python解法

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

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

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

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

例:

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

解析:

我们可以将每天的状态分为四种,分别为0已购买状态(购买当天或购买后没有任何操作),1卖出股票,并且度过冷冻期后一直无操作状态,2卖出状态(只有一天,因为第二天就是冷冻期状态),3冷冻期状态。分别用0,1,2,3来表示。每一天都有可能对应着不同的状态,所以我们用横坐标代表天数,纵坐标代表状态创建dp数组。

成为状态0之前可能是状态0,状态1,状态3。因为状态2的第二天是冷冻期(状态3)所以只有三种情况。

成为状态1之前可能是状态1,状态3。

成为状态2之前只能是状态0。

成为状态3之前只能是状态2。

至于所有状态的初始值,除了买入状态0需要扣除买入资金,其余的收益均为0。

class Solution(object):
    def maxProfit(self, prices):
        """
        :type prices: List[int]
        :rtype: int
        """
        n = len(prices)
        if n == 0:  # 边界判断
            return 0

        dp = [[0] * 4 for _ in range(n)]  # 初始化dp数组
        dp[0][0] = -prices[0]  # 状态0初始化
        for i in range(1, n):  # 天数遍历
            dp[i][0] = max(dp[i - 1][0], max(dp[i - 1][3], dp[i - 1][1]) - prices[i])  # 成为状态0的算法思路
            dp[i][1] = max(dp[i - 1][1], dp[i - 1][3])  # 成为状态1
            dp[i][2] = dp[i - 1][0] + prices[i]  # 成为状态2
            dp[i][3] = dp[i - 1][2]  # 成为状态3
        return max(dp[n - 1][3], dp[n - 1][1], dp[n - 1][2])  # 状态0需要购入,所以无需考虑

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值