代码随想录算法训练营第42天|188.买卖股票的最佳时机IV、309.最佳买卖股票时机含冷冻期、714.买卖股票的最佳时机含手续费

1.188.买卖股票的最佳时机IV

题目链接：188.买卖股票的最佳时机IV
文档讲解： 代码随想录

（1）确定数组和下标
用一个二维数组dp[i][j]来表示状态，在第 i 天的状态为 j，所剩下的最大金额，j 的状态表示为：
0：不操作
1：第1次持有
2：第1次不持有
3：第2次持有
4：第2次不持有
……
j 的范围定义为 2k + 1
（2）递推关系式
dp[i][1]：第 i 天买入股票，那么则为dp[i-1][0]- prices[i]；在之前就买入，则为dp[i-1][1]；
dp[i][2]：第 i 天卖出股票，那么则为dp[i-1][1] + prices[i]；在之前就卖出，则为dp[i-1][2];
类比可得，j 从0开始
dp[i][j+1] = max(dp[i-1][j] - prices[i],dp[i-1][j+1])
dp[i][j+2] = max(dp[i-1][j+1]+prices[i],dp[i-1][j+2])
（3）初始化
dp[i][j为奇数] = -prices[0],dp[0][j为偶数] = 0；其余值在定义数组的时候值为0
（4）遍历顺序
从前往后，使用两层循环，一层遍历i，一层遍历m
（5）打印数组

class Solution(object):
    def maxProfit(self, k, prices):
        """
        :type k: int
        :type prices: List[int]
        :rtype: int
        """
        dp = [[0] * (2*k+1) for _ in prices]
        for j in range(1,2*k,2):
            dp[0][j] = -prices[0]
        for i in range(1,len(prices)):
            for j in range(0,2*k-1,2):
                dp[i][j+1] = max(dp[i-1][j] - prices[i],dp[i-1][j+1])
                dp[i][j+2] = max(dp[i-1][j+1] + prices[i],dp[i-1][j+2])
        return max(dp[-1][-1],dp[-1][-2])

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

题目链接：309.最佳买卖股票时机含冷冻期
文档讲解： 代码随想录

（1）确定数组和下标
dp[i][j]表示第 i 天状态 j 的最大金额，j 包含：
0：持有股票
1：保持卖出股票
2：今天卖出股票
3：冷冻期
（2）确定递推关系式
dp[i][0]：之前买入，dp[i-1][0]；今天买入，可以由0和3状态得到，dp[i-1][1] - prices[i]或者dp[i-1][3] - prices[i]
dp[i][1]：dp[i-1][1]或者dp[i-1][3]
dp[i][2]：dp[i-1][0] + prices[i]
dp[i][3]：dp[i-1][2]
（3）初始化
dp[0][0]=-prices[0],dp[0][1]=0,dp[0][1]=0,dp[0][2]=0
（4）遍历顺序：从前往后
（5）打印数组

class Solution(object):
    def maxProfit(self, prices):
        """
        :type prices: List[int]
        :rtype: int
        """
        dp = [[0] * 4 for _ in prices]
        dp[0][0] = -prices[0]
        for i in range(1, len(prices)):
            dp[i][0] = max(dp[i-1][0], dp[i-1][1] - prices[i], dp[i-1][3] - 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[-1][0],dp[-1][1],dp[-1][2],dp[-1][3])

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

题目链接：714.买卖股票的最佳时机含手续费
文档讲解： 代码随想录

（1）确定数组和下标
dp[i][j]表示第 i 天状态为 j 所获得的最大金额，j 有两种，0为持有股票，1为不持有股票
（2）递推关系式
dp[i][0]：之前买入，dp[i-1][0]；本轮买入，dp[i-1][1] - prices[i]
dp[i][1]：之前卖出，dp[i-1][1]；本轮卖出，dp[i-1][0] + prices[i] - fee
（3）初始化
dp[0][0] = -prices[0], dp[0][1] = 0
（4）遍历顺序：从前往后
（5）打印数组

class Solution(object):
    def maxProfit(self, prices, fee):
        """
        :type prices: List[int]
        :type fee: int
        :rtype: int
        """
        if len(prices) == 1:
            return 0

        dp = [[0] * 2 for _ in prices]
        dp[0][0] = -prices[0]
        for i in range(1, len(prices)):
            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 max(dp[-1][0],dp[-1][1])

总结：如果有买卖次数限制，有不操作状态，若无，则不用。