利用python 完成 leetcode123 买卖股票的最佳时机 III

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

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

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

示例 1:

输入: [3,3,5,0,0,3,1,4]
输出: 6
解释: 在第 4 天（股票价格 = 0）的时候买入，在第 6 天（股票价格 = 3）的时候卖出，这笔交易所能获得利润 = 3-0 = 3 。
随后，在第 7 天（股票价格 = 1）的时候买入，在第 8 天 （股票价格 = 4）的时候卖出，这笔交易所能获得利润 = 4-1 = 3 。
示例 2:

输入: [1,2,3,4,5]
输出: 4
解释: 在第 1 天（股票价格 = 1）的时候买入，在第 5 天 （股票价格 = 5）的时候卖出, 这笔交易所能获得利润 = 5-1 = 4 。
注意你不能在第 1 天和第 2 天接连购买股票，之后再将它们卖出。
因为这样属于同时参与了多笔交易，你必须在再次购买前出售掉之前的股票。
示例 3:

输入: [7,6,4,3,1]
输出: 0
解释: 在这个情况下, 没有交易完成, 所以最大利润为 0。
思路
股票可以当天买并在当天卖出
用l1储存每一天将买到的第一支股票卖出后能剩下最多的钱
对于第i天，能剩下最多的钱等于用第i天的股票价格减去前i天股票最便宜的价格
s1记录前i天，买到最便宜的价格股票后剩下的钱
用l2储存第二支股票卖出后总共能剩下最多的钱
计算方法同上
代码

def maxProfit(self, prices: List[int]) -> int:
    if len(prices)<=1:return 0
    l1=[0]*len(prices)
    l2=[0]*len(prices)
    s1=-prices[0]
    for i in range(1,len(prices)):
        s1=max(s1,-prices[i])
        l1[i]=max(l1[i-1],s1+prices[i])#如果能挣更多的钱则在今天卖出
    s2=-prices[0]
    for i in range(1,len(prices)):
        s2=max(s2,l1[i]-prices[i])
        l2[i]=max(l2[i-1],s2+prices[i])
    return l2[-1]

其实对于每一天的计算只要靠前一天的结果就行了。

def maxProfit(self, prices: List[int]) -> int:
        if len(prices)<=1:return 0
        s1,s2=0,0
        b1,b2=-prices[0],-prices[0]

        for i in prices[1::]:
            b1=max(b1,-i);
            s1=max(s1,b1+i);
            b2=max(b2,s1-i);
            s2=max(s2,b2+i);

        return s2;