这题可以用贪心算法。
题目
给定一个数组 prices
,它的第 i
个元素 prices[i]
表示一支给定股票第 i
天的价格。
你只能选择 某一天 买入这只股票,并选择在 未来的某一个不同的日子 卖出该股票。设计一个算法来计算你所能获取的最大利润。
返回你可以从这笔交易中获取的最大利润。如果你不能获取任何利润,返回 0
。
思路
我先用的循环遍历去解,能过示例,但是提交不通过,最后用贪心算法通过。
贪心算法的解法思路:
- 遍历价格数组:记录遍历到当前天的最低价格,并计算当前价格与最低价格之间的差值,记录最大的差值作为最大利润。
- 更新历史最低价格:每遇到一个比当前最低价格更低的价格时,更新最低价格。
- 计算利润:每次计算当前价格与历史最低价格之间的差值,若差值大于之前记录的最大利润,则更新最大利润。
代码
class Solution:
def maxProfit(self, prices: List[int]) -> int:
# 初始化最低价格为无穷大,最大利润为0
min_price = float('inf')
max_profit = 0
# 遍历价格数组
for price in prices:
# 如果当前价格低于最低价格,则更新最低价格
if price < min_price:
min_price = price
# 否则,计算当前价格与最低价格的差值,更新最大利润
elif price - min_price > max_profit:
max_profit = price - min_price
# 返回最大利润
return max_profit