一、任务
给定一个数组 prices
,它的第 i
个元素 prices[i]
表示一支给定股票第 i
天的价格。
你只能选择 某一天 买入这只股票,并选择在 未来的某一个不同的日子 卖出该股票。设计一个算法来计算你所能获取的最大利润。
返回你可以从这笔交易中获取的最大利润。如果你不能获取任何利润,返回 0
。
二、思路
1. 暴力求解(超时)
遍历价格中的元素,及该元素后最大的元素,计算利润,更新最大利润
问题:每次计算最大值都需要遍历一次,如果改为复用就能减少时间
2. 优化
遍历价格中的元素,以及该元素前的最小元素,计算利润,更新最大利润
三、解答
class Solution:
def maxProfit(self, prices: List[int]) -> int:
profit = 0
min_val = prices[0]
for i,item in enumerate(prices[1:]):
if item < min_val:
min_val = item
else:
new_profit = item - min_val
if new_profit > profit:
profit = new_profit
return profit