代码随想录27期|Python|Day48|​动态规划|​ 188.买卖股票的最佳时机IV|309.最佳买卖股票时机含冷冻期|714.买卖股票的最佳时机含手续费

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

本题在昨天的III的基础上更进一步，考察递推公式的配对更新。

这里着重讲递推细节。

在III的基础上，将每个时刻的状态扩充为（2k+1）种，初始化为0 。然后将奇数索引的值全部设置为-prices[0]，偶数不变。

在更新过程中，我们发现，对于每个时刻而言，除了状态0（没有操作，默认为0）以外，每次买入都对应一次卖出，也就是更新是成对更新的。

为什么？因为dp储存的是最优解，那么有买入一定会有卖出（不然亏损不可能最优）。

因此更新的递推公式为：

dp[i][j+1] = max(dp[i-1][j+1], dp[i-1][j]-prices[i]) // 第j次买入
dp[i][j+2] = max(dp[i-1][j+2], dp[i-1][j+1] + prices[i])  //第j次卖出

然后看一下更新的时候j的写法：

状态值的索引是从0一直可以取到2k的，而且以2个为一组（0状态固定不变不跟新）。

以第1对状态为例：所以此处状态的开始值是0（j），更新1（j+1），更新2（j+2），最后终结在2k-1处（考虑索引范围，j + 2 < 2k+1）。

class Solution(object):
    def maxProfit(self, k, prices):
        """
        :type k: int
        :type prices: List[int]
        :rtype: int
        """
        dp = [[0] * (2 * k + 1)] * len(prices)

        for j in range(1, 2 * k + 1, 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+1], dp[i-1][j]-prices[i])
                dp[i][j+2] = max(dp[i-1][j+2], dp[i-1][j+1] + prices[i])
        return max(dp[-1])

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

本题的状态更加细分，由于有冷静期，所以不持有股票的状态更加复杂，可以分为以下4个状态。

本题的关键就在于dp数组的含义和递推公式的写法。

1、dp数组的含义

一共4个状态：

（1）今天持有股票

（2）今天刚卖出 

（3）今天是冷静期

（4）今天是过了冷静期的不持有状态

2、递推更新公式

（1）今天持有 = max（昨天持有，昨天是冷静期-今日价格，昨天是过了冷静期的不持有-今日价格）

（2）今天卖出 = 昨天持有 + 今日价格

（3）今天是冷静期 = 昨天卖出

（4）今天是已经过了冷静期的不持有状态 = max（昨天是过了冷静期的不持有状态，昨天是冷静期）

class Solution(object):
    def maxProfit(self, prices):
        """
        :type prices: List[int]
        :rtype: int
        """
        dp = [[0] * 4 for _ in range(len(prices))]

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

        for i in range(1, len(prices)):
            dp[i][0] = max(dp[i-1][0], max(dp[i-1][3], dp[i-1][2]) - prices[i])  # 今天持有股票
            dp[i][1] = dp[i-1][0] + prices[i]  # 今天卖出
            dp[i][2] = dp[i-1][1]  # 今天是冷却期
            dp[i][3] = max(dp[i-1][3], dp[i-1][2])  # 已卖出（过了冷却期）

        return max(dp[-1])

 此外，这里还想讨论一下关于创建dp数组的语法的区别：（之前都没注意，不知道咋过的）

1、a = [[0] * 4] * 4 

本意是创建一个一维数组然后复制4次，所以他们都指向同一个地址，改变一个元素会改变全部的对应位置处的元素。

2、a = [[0] * 4 for _ in range(4)]

循环创建一个2维数组，每一个数组是独立的，一定采用这样的写法。

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

本题和之前的不限制次数买股票其实是一样的，只不过是在卖出的时候需要多扣掉一笔手续费fee。

为什么是在卖出的时候扣掉？

买卖只要支付一次手续费，肯定是在卖出的时候才能比较净利润的多少。如果扣在了买入，就会影响max操作。

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

买卖股票问题总结

代码随想录

Day48完结！！！