代码随想录算法训练营四十一天|188.买卖股票的最佳时机IV、714.买卖股票的最佳时机含手续费

题目链接:：188. 买卖股票的最佳时机 IV - 力扣（LeetCode）

思路：跟两次那题思路类似，将两天扩大到了n天

class Solution(object):
    def maxProfit(self, k, prices):
        """
        :type k: int
        :type prices: List[int]
        :rtype: int
        """

        if len(prices) == 0:
            return 0

        dp = [[0] * (2*k + 1) for _ in range(len(prices))]

        for j in range(1, 2*k, 2): #k = 2 ; 2k + 1 = 5 = 0, 1, 2, 3, 4 ; j取 1, 3
            dp[0][j] = -prices[0]

        for i in range(1, len(prices)):
            for j in range(0, 2*k-1, 2):# 1,2, j = 0 ; 3,4; j = 2 #第i天买入, 第j次买入,第j次卖出
                dp[i][j+1] = max(dp[i-1][j+1], dp[i-1][j] - prices[i]) #要么遵循i-1天的j+1次买入的值，要么遵循i-1天卖出之后的利润减去新买的股票的值
                dp[i][j+2] = max(dp[i-1][j+2], dp[i-1][j+1] + prices[i]) #要么遵循i-1天的i+2卖出的值，要么遵循i-1天买入股票之后的钱加上卖出股票所得的总利润

        
        return dp[-1][2*k]

题目链接：309. 买卖股票的最佳时机含冷冻期 - 力扣（LeetCode）

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)]
        # 4个状态分别表示
        # 1.持有股票、
        # 2.不持有股票且处于冷冻期、
        # 3.不持有股票且不处于冷冻期、
        # 4.不持有股票且当天卖出后处于冷冻期
        dp[0][0] = -prices[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])  # 当前持有股票的最大利润等于前一天持有股票的最大利润或者前一天不持有股票且不处于冷冻期的最大利润减去当前股票的价格
            dp[i][1] = max(dp[i-1][1], dp[i-1][3])  # 当前不持有股票且处于冷冻期的最大利润等于前一天持有股票的最大利润加上当前股票的价格
            dp[i][2] = dp[i-1][0] + prices[i]  # 当前不持有股票且不处于冷冻期的最大利润等于前一天不持有股票的最大利润或者前一天处于冷冻期的最大利润
            dp[i][3] = dp[i-1][2]  # 当前不持有股票且当天卖出后处于冷冻期的最大利润等于前一天不持有股票且不处于冷冻期的最大利润
        return max(dp[n-1][3], dp[n-1][1], dp[n-1][2])  # 返回最后一天不持有股票的最大利润

题目链接：714. 买卖股票的最佳时机含手续费 - 力扣（LeetCode）

思路：122 题目的基础上 交易的时候减去一笔手续费即可

class Solution(object):
    def maxProfit(self, prices, fee):
        """
        :type prices: List[int]
        :type fee: int
        :rtype: int
        """

        size = len(prices)

        dp = [[0] * 2 for _ in range(size)]

        dp[0][0] -= prices[0]
        dp[0][1] = 0

        for i in range(1, size):
            dp[i][0] = max(dp[i - 1][0], dp[i - 1][1] - prices[i])
            dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] + prices[i] - fee)


        return dp[size - 1][1]