python_leetcode121. 买卖股票的最佳时机

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

如果你最多只允许完成一笔交易(即买入和卖出一支股票),设计一个算法来计算你所能获取的最大利润。

注意你不能在买入股票前卖出股票。

示例 1:

输入: [7,1,5,3,6,4]
输出: 5
解释: 在第 2 天(股票价格 = 1)的时候买入,在第 5 天(股票价格 = 6)的时候卖出,最大利润 = 6-1 = 5 。
     注意利润不能是 7-1 = 6, 因为卖出价格需要大于买入价格。

示例 2:

输入: [7,6,4,3,1]
输出: 0
解释: 在这种情况下, 没有交易完成, 所以最大利润为 0。

偷懒想法:遍历两次

class Solution(object):
    def maxProfit(self, prices):
        """
        :type prices: List[int]
        :rtype: int
        """
        n = len(prices)
        mp = 0
        for i in range(n-1):
            for j in range(i+1, n):
                profit = prices[j]-prices[i]
                if profit > 0:
                    mp = max(mp, profit)
        return mp

结果发现:
        

于是遍历一次:用prices[0]为第一次的最小值,遍历取最大利润,如果有比prices[0]更小的直接替换,不用重新遍历。

class Solution(object):
    def maxProfit(self, prices):
        """
        :type prices: List[int]
        :rtype: int
        """
        if len(prices)<2:
            return 0
        max_profit = 0
        min_price = prices[0]
        for i in price:
            min_price=min(min_price,i)
            max_profit=max(max_profit, i-min_price)
        return max_profit
        

优化下代码:

class Solution(object):
    def maxProfit(self, prices):
        """
        :type prices: List[int]
        :rtype: int
        """
        if prices==[]:
            return 0
        max_profit = 0
        min_price = prices[0]
        for i in prices[1:]:
            if i < min_price:
                min_price=i
            max_profit=max(max_profit, i-min_price)
        return max_profit
        

猜你喜欢

转载自blog.csdn.net/AntiZheng/article/details/82628440
今日推荐