买卖股票的最佳时机三

一曲冷凌霜 提交于 2020-03-07 07:08:16

给定一个数组,它的第 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。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-iii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

class Solution(object):
    def maxProfit(self, prices):
        """
        :type prices: List[int]
        :rtype: int
        """
        res=0
        n=len(prices)
        if n<2:
            return 0
        left=[0 for _ in range(n)]
        min_value=prices[0]
        for i in range(n):
            left[i]=max(left[i-1],prices[i]-min_value)
            min_value=min(min_value,prices[i])
        right=[0 for _ in range(n) ]
        max_value=prices[n-1]
        for i in range(n-1)[::-1]:
            right[i]=max(right[i+1],max_value-prices[i])
            max_value=max(max_value,prices[i])
        res=max(left[n-1],right[0])
        for i in range(n-2):
            # res=max(left[i],right[i+1])
            res=max(res,left[i]+right[i+1])
        return res

这道题一因为它限制了两次交易,所以思路是遍历一边数组对于每一个数字都分别去查找他左右两边的最大的利润,然后遍历一下求出最大的两次交易的利润。
这个地方还有一个小问题就是可能存在有一次交易比两次交易利润大的情况,我们只需要单独再算一下一次交易的利润然后取最大即可。

class Solution(object):
    def maxProfit(self, prices):
        """
        :type prices: List[int]
        :rtype: int
        """
        n=len(prices)
        if n<2:
            return 0
        in1=-prices[0]
        out1=0
        in2=float('-inf')
        out2=0
        for i in range(n):
            out2=max(prices[i]+in2,out2)
            in2=max(out1-prices[i],in2)
            out1=max(prices[i]+in1,out1)
            in1=max(-prices[i],in1) 
        return max(out2,out1)

还有一种思路就是对于每一天我们其实都是在五种状态的可能
啥都没买
第一次买了股票后
第一次卖出股票后
第二次买了股票后
第二次卖出股票后
我们要做的就是对每一天的每种状态的进行更新,取其最大值即可
在这里我做了一个优化就是让后四种状态的初始值是0,这样最大利润永远是大于等于0的所以,啥都没买的情况就可以不考虑了。

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!