121. 买卖股票的最佳时机
题目 121. 买卖股票的最佳时机
给定一个数组 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。
思路
- 方法一:双循环,但是会超时
- 方法二:
- 假设我们会在第i天卖股票时能赚多少钱
- 我们只要用一个变量记录一个历史最低价格 minprice。那么我们在第 i 天卖出股票能得到的利润就是 prices[i] - minprice。
- 我们只需要遍历价格数组一遍,记录历史最低点,然后在每一天考虑一个问题:如果我是在历史最低点买进的,那么我今天卖出能赚多少钱?当考虑完所有天数之时,我们就得到了最好的答案。
代码
# # 此方法会超时
# class Solution:
# def maxProfit(self, prices: List[int]) -> int:
# res = 0
# for i in range(len(prices)):
# for j in range(i + 1, len(prices)):
# res = max(res, prices[j] - prices[i])
# return res
# 方法二 假设设置一个历史最低价格
class Solution:
def maxProfit(self, prices: List[int]) -> int:
res = 0
minprofit = inf # 历史最低价格
for i in prices:
res = max(i-minprofit, res)
minprofit = min(minprofit, i)
return res