给定一个数组,它的第 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。
链接:https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock
首先想到的是暴力遍历,n方复杂度,轻轻松松地超时了:
class Solution:
def maxProfit(self, prices: List[int]) -> int:
res = 0
length = len(prices)
for i in range(length-1):
for j in range(i+1, length):
diff = prices[j]-prices[i]
if diff > res:
res = diff
return res
接下来我实在想不出来了,就看了标答,果然有一种n复杂度的方法。
思路是从前往后依次遍历每一个元素,目标是找到最小的低谷之后的最大的峰值,它们的差值就是最大的利润。
对于遍历到的每个元素,都要判别:
- 它是不是遍历到现在为止的最小值
- 如果不是最小值,那么它是不是遍历到现在为止,出现在最小值后面的最大值
看标答里配的那个折线图,真的有助于让思路清晰。
class Solution:
def maxProfit(self, prices: List[int]) -> int:
res = 0
minprice = float('inf')
for i in range(len(prices)):
if prices[i] < minprice:
minprice = prices[i]
elif prices[i]-minprice > res:
res = prices[i]-minprice
return res