day51|买卖股票问题5-6(冷冻期+手续费)

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

含有冷冻期之后,需要dp数组再多加一个状态即冷冻期,如果这一天是冷冻期的最大收益是多少。

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        # 不处理,持有,不持有,冷冻期(就是不处理)
        N = len(prices)
        dp = [[0]*4 for i in range(N)]
        dp[0][0] = 0
        dp[0][1] = -prices[0]
        dp[0][2] = 0
        dp[0][3] = 0
        for i in range(1,N):
            # 持有股票
                # 如果此前就持有该股票:dp[i-1][1]
                # 如果此前未持有该股票新购入,需要满足前一天冷冻期,今天买入;继续保持前一天持有状态
            dp[i][1] = max(dp[i-1][3] - prices[i] ,dp[i-1][1])
            dp[i][2] = max(dp[i-1][1] + prices[i] ,dp[i-1][2])
            dp[i][3] = dp[i-1][2]
        return max(dp[N-1])

714.买卖股票的最佳时机含手续费

类似于买卖股票II,在卖出股票的时候将手续费减去即可。

class Solution:
    def maxProfit(self, prices: List[int], fee: int) -> int:
        N = len(prices)
        dp = [[0,0] for _ in range(N)]
        dp[0][0] = -prices[0] 
        dp[0][1] = 0
        for i in range(1,N):
            # 0表示不持有该股票;1表示持有该股票
            # 如果不持有该股票,一种情况是买入,另一种情况是继续保持不持有的状态
            # 如果当前是持有该股票的状态,一种情况是卖出,另一种情况是继续保持之前的持有状态,不卖出
            dp[i][0] = max(dp[i-1][1]-prices[i],dp[i-1][0])
            dp[i][1] = max(dp[i-1][1],dp[i-1][0] + prices[i] - fee)
        return max(dp[N-1][0],dp[N-1][1]) 

总结股票问题

day51|买卖股票问题5-6(冷冻期+手续费)_第1张图片

你可能感兴趣的:(动态规划)