持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第8天,点击查看活动详情
题目描述
原题链接 :
给定一个数组 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
思路分析
思路:使用一个单调递增栈实现。具体做法如下:
- 遍历数组prices,如果栈顶元素小于当前遍历的元素,则当前元素直接入栈;如果栈顶元素大于当前遍历的元素,栈顶元素出栈,直到栈顶元素不大于当前遍历元素为止,这个时候当前元素才能入栈。
- 栈顶元素 减 栈底元素,就是当前所遍历到的结果的最优解,记录下来。
- 返回结果。
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