【Leetcode】121题买卖股票Python实现

题目描述

121给定一个数组 prices ，它的第 i 个元素 prices[i] 表示一支给定股票第 i 天的价格。

你只能选择 某一天 买入这只股票，并选择在 未来的某一个不同的日子 卖出该股票。设计一个算法来计算你所能获取的最大利润。

返回你可以从这笔交易中获取的最大利润。如果你不能获取任何利润，返回 0 。

示例

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

方法总结

用一个数pmin记录前i个中最小的数。然后用第i个数与pmin做差，和上一次最大获利值max_val做比较，得出最新的最大获利值。

总代码（记录最小值版本）

class Solution(object):
    def maxProfit(self, prices):
        """
        :type prices: List[int]
        :rtype: int
        """
        max_val = 0
        pmin = 10000
        for i in range(len(prices)):
            pmin = min(pmin, prices[i])
            max_val = max(prices[i] - pmin, max_val)
        return  max_val

总代码2（记录最大值版本，超出时间限制）

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

分析

在找最大值版本的代码中，max函数运算比较的值比在找最小值版本的代码中时间花费要大。
最小值版本：只比较两个数的大小，取小。
而最大值版本：寻找i+1到n中最大的值，因此超出时间限制。当然也可以从后向前，类似反向的最小值求法。