代码随想录Day 43|Leetcode|Python|123. 买卖股票的最佳时机III，188.买卖股票的最佳时机IV

123.买卖股票的最佳时机III

给定一个数组，它的第 i 个元素是一支给定的股票在第 i 天的价格。

设计一个算法来计算你所能获取的最大利润。你最多可以完成 两笔 交易。

注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。

解题思路：

确认dp数组含义：

dp[i][0]第i天没有进行任何操作的最大金额数

dp[i][1]第i天第一次持有股票时的金额数

dp[i][2]第i天第一次不持有股票时的金额数

dp[i][3]第i天第二次持有股票时的金额数

dp[i][4]第i天第二次不持有股票时的金额数

dp公式：dp[i][1] = max(dp[i-1][1], dp[i-1][0]-prices[i])
            dp[i][2] = max(dp[i-1][2], dp[i-1][1]+prices[i])
            dp[i][3] = max(dp[i-1][3], dp[i-1][2]-prices[i])
            dp[i][4] = max(dp[i-1][4], dp[i-1][3]+prices[i])

初始化：dp[0][0] = 0, dp[0][1] = -prices[0], dp[0][2] = 0在第一天买入又买出，

dp[0][3] = -prices[0], dp[0][4] = 0

遍历顺序：从前向后遍历

打印dp数组：

以输入[1,2,3,4,5]为例，红色为最后两次卖出状态

代码如下：

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        dp = [[0]*5 for _ in range(len(prices))]
        # dp[0][0] = 0
        dp[0][1] = -prices[0]
        dp[0][2] = 0
        dp[0][3] = -prices[0]
        dp[0][4] = 0
        for i in range(1, len(prices)):
            # dp[i][0] = 0
            dp[i][1] = max(dp[i-1][1], dp[i-1][0]-prices[i])
            dp[i][2] = max(dp[i-1][2], dp[i-1][1]+prices[i])
            dp[i][3] = max(dp[i-1][3], dp[i-1][2]-prices[i])
            dp[i][4] = max(dp[i-1][4], dp[i-1][3]+prices[i])
        dp1 = dp[len(prices)-1][2]
        dp2 = dp[len(prices)-1][4]
        return max(dp1, dp2)

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

给你一个整数数组 prices 和一个整数 k ，其中 prices[i] 是某支给定的股票在第 i 天的价格。

设计一个算法来计算你所能获取的最大利润。你最多可以完成 k 笔交易。也就是说，你最多可以买 k 次，卖 k 次。

注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。

解题思路：

本题与上一题区别在于买卖次数不确定，是k次上一题是两次，因此会有两次买入和卖出，用

dp[i][1], dp[i][2], dp[i][3], dp[i][4] 表示。因此对于k次买卖，总共操作次数是2k，加上第零次是2k+1个index数。对于每次操作，用 j， j+1， j+2表示，j对应第零次操作，j+1对应该组第一次即买入操作，j+2对应该组第二次即卖出操作。

确认dp数组含义：

dp[i][j]第i天没有进行任何操作的最大金额数

dp[i][j+1]第i天第一次持有股票时的金额数

dp[i][j+2]第i天第一次不持有股票时的金额数

初始化：dp[0][j] = 0, dp[0][j+1] = -prices[0], dp[0][j+2] = 0在第一天买入又买出

dp公式: while loop j starts with 1, j += 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])

遍历顺序：从前向后遍历

打印dp数组：

以输入[1,2,3,4,5]，k=2为例，红色部分为最后求解

代码如下：

class Solution:
    def maxProfit(self, k: int, prices: List[int]) -> int:
        dp = [[0]*(2*k+1) for _ in range(len(prices))]
        j = 1
        while j<2*k:
            dp[0][j] = -prices[0]#initialize
            j += 2
        for i in range(1, len(prices)):
            #每一天都进行不同买卖次数衡量
            j = 0
            while j < 2*k:
                #买入后的金额
                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])
                j += 2
        #比较倒数第二次卖出和最后一次卖出后的金额
        dp1 = dp[len(prices)-1][2*(k-1)+2]
        dp2 = dp[len(prices)-1][2*k]
        return max(dp1, dp2)