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

dp定义为dp[i][k][0]第i天没有持股，恰好完成了k笔交易（卖出时算一笔交易完成）的情况下可以获得的最大利润
dp[0][k>=1][0or1] 不合法，因为是第一天不可能进行过交易
这题就是k=2
注意按照这个通解，由于定义的是恰好完成k笔交易在卖出时算交易完成，根据递推公式无法涉dp[i][0][1]
所以这一步放在k的循环之外

def maxProfit(self, k, prices):
        """
        :type k: int
        :type prices: List[int]
        :rtype: int
        """
        #dp定义为dp[i][k][0]第i天没有持股，恰好完成了k笔交易（卖出时算一笔交易完成）的情况下可以获得的最大利润
        #dp[0][k>=1][0or1] 不合法，因为是第一天不可能进行过交易
        n = len(prices)
        k = min(k,n//2)
        res = 0
        dp = [[[0]*2 for j in range(k+1)] for i in range(n)]
        dp[0][0][1] = - prices[0]
        for j in range(1,k):
            dp[0][j][0] = -0x3f3f3f3f
            dp[0][j][1] = -0x3f3f3f3f 
        for i in range(1,n):
            dp[i][0][1] = max(dp[i - 1][0][1], dp[i - 1][0][0] - prices[i])  #第i天进行了0次交易持有股票
            for j in range(k,0,-1):                                          #持有股票在0次交易的时候没有考虑到
                dp[i][j][0] = max(dp[i-1][j][0],dp[i-1][j-1][1] + prices[i])
                dp[i][j][1] = max(dp[i-1][j][1],dp[i-1][j][0] - prices[i])
        for i in range(1,k+1):
            res = max(res,dp[n-1][i][0])
        return res 

def maxProfit(self, prices):
    """
    :type prices: List[int]
    :rtype: int
    """
    n = len(prices)
    dp = [[0]*4 for i in range(n)] #四个状态分别代表第一次买入，第一次卖出，第二次买入，第二次卖出
    dp[0][0] = -prices[0]
    dp[0][2] = -prices[0]     #注意这里
    for i in range(1,n):
        dp[i][0] = max(dp[i-1][0],-prices[i])
        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])
    return max(dp[n-1][1],dp[n-1][3])