买卖股票的最佳时机(迭代判断)
1.题目描述
- 给定一个数组 prices ,它的第 i 个元素 prices[i] 表示一支给定股票第 i 天的价格。
- 你只能选择 某一天 买入这只股票,并选择在 未来的某一个不同的日子 卖出该股票。设计一个算法来计算你所能获取的最大利润。
- 返回你可以从这笔交易中获取的最大利润。如果你不能获取任何利润,返回 0
2.输入输出案例
输入:[7,1,5,3,6,4]
输出:5
输入:prices = [7,6,4,3,1]
输出:0
3.解题思路(语言不限C++/C/java/python均可)
- 对于问题和输入,不难发现,它其实是一个找最大最小值的问题,但是它给出了一个限制,即最大值必须在最小值的后面才有意义(可以低价买到股票,但必须在后面的日期卖出)。
- 对于输入 [7,1,5,3,6,4] 我们可以,利用迭代遍历的方法找出最大与最小值。
- 然后逐步迭代,如果当前数组元素值小于最小值,最小值以及最小值的位置均更新。
- 如果当前数组元素值大于最小值且当前数组值减去最小值获得的利润大于等于之前的利润(初始为0)则更新利润。
- 输入[7,6,4,3,1]大家可以自行分析,图示如下:
4.代码实现
def maxProfit(self, prices: List[int]) -> int:
# 初始化最大、最小值、最大值最小值对应的位置
if not prices or len(prices) == 1:
return 0
min_v = prices[0]
min_pos = 0
val = 0
idx = 1
while idx < len(prices):
if min_v > prices[idx] and idx != (len(prices) - 1):
min_v = prices[idx]
min_pos = idx
continue
elif min_v < prices[idx]:
c_val = prices[idx] - min_v
if c_val >= val:
val = c_val
idx += 1
# 找到最大最小值,判断是否满足题给限制
return val