力扣:121. 买卖股票的最佳时机(Python3)

题目:

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

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

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

来源:力扣(LeetCode)
链接:力扣(LeetCode)官网 - 全球极客挚爱的技术成长平台

示例:

示例 1:

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


示例 2:

输入:prices = [7,6,4,3,1]
输出:0

解释:在这种情况下, 没有交易完成, 所以最大利润为 0。

解法:

遍历prices,去除连续重复元素。

遍历prices,记录到每个值为止的最小值。

反向遍历prices,记录到每个值为止最大的值,记录的时候每次在列表头部插入。

把上述2个记录中的值对应相减,返回最大值。

知识点:

1.zip([iterable, ...]):iterable表示一个或多个迭代器,将可迭代的对象作为参数,将对象中对应的元素打包成一个个元组,返回元祖列表,而且元素个数与最短的列表一致。例如:a = [1, 2, 3], b = [4, 5, 6], 输出list(zip(a, b))为[(1, 4), (2, 5), (3, 6)]。

代码:

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        new = [prices[0]]
        if len(prices) > 1:
            for num in prices[1:]:
                if num != new[-1]:
                    new.append(num)
        prices = new
        low = []
        high = []
        price = prices[0]
        for num in prices:
            if num < price:
                price = num
            low.append(price)
        price = prices[-1]
        for num in prices[::-1]:
            if num > price:
                price = num
            high.insert(0, price)
        return max(map(lambda x: x[1] - x[0], zip(low, high)))

你可能感兴趣的:(LeetCode,leetcode,算法,python)