题解 | #买卖股票的最好时机(三)#

#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# 两次交易所能获得的最大收益
# @param prices int整型一维数组 股票每一天的价格
# @return int整型
#
class Solution:
    def maxProfit(self , prices: List[int]) -> int:
        # write code here
        length = len(prices)
        if length <5:
            return 0
        dp = [[0]*5]*(length)
        dp[0][0] = 0
        dp[0][1] = -prices[0]#初始化：买过一次股票还没有卖出，初始为-prices[0]
        dp[0][3] = -prices[0]#初始化：买过两次只卖出过一次股票，初始为-prices[0]
        for i in range(1,length):
            dp[i][0] =  dp[i-1][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])
        return dp[length-1][4]