力扣题解:121. 买卖股票的最佳时机

87 阅读1分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第8天,点击查看活动详情

题目描述

原题链接 :

121. 买卖股票的最佳时机 - 力扣(LeetCode)

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

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

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

 

示例 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。

提示:

  • 1 <= prices.length <= 10^5
  • 0 <= prices[i] <= 10^4

思路分析

思路:使用一个单调递增栈实现。具体做法如下:

  1. 遍历数组prices,如果栈顶元素小于当前遍历的元素,则当前元素直接入栈;如果栈顶元素大于当前遍历的元素,栈顶元素出栈,直到栈顶元素不大于当前遍历元素为止,这个时候当前元素才能入栈。
  2. 栈顶元素 减 栈底元素,就是当前所遍历到的结果的最优解,记录下来。
  3. 返回结果。

AC 代码

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        stack = []
        res = 0 # 记录最大值
        for i in range(len(prices)):
            while stack:
                if prices[i] < stack[-1]:
                    stack.pop()
                else:
                    break
            stack.append(prices[i])
            if res < stack[-1] - stack[0]:
                res = stack[-1] - stack[0]
        return res

参考

dp,最大价格=(前一天价格减去之前的最小价格,之前的最大价格) - 买卖股票的最佳时机 - 力扣(LeetCode)