LeetCode Top100: 买卖股票的最佳时机 (python)

发布时间 2023-04-20 16:31:30作者: 华东博客

LeetCode Top100: 买卖股票的最佳时机

 

给定一个数组 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。

 

提示:

  • 1 <= prices.length <= 105
  • 0 <= prices[i] <= 104

 

实现:

以下是一个Python程序,使用一次遍历来计算最大利润。在遍历过程中,记录当前最低价格和最大利润,并更新它们以获取最大利润。

使用一个变量 min_price 来记录到目前为止的最低价格,以及一个变量 max_profit 来记录到目前为止的最大利润。在遍历股票价格数组时,如果当前价格小于最低价格,则更新最低价格;否则,计算当前价格与最低价格的差价,并将其与最大利润进行比较,更新最大利润。最终返回最大利润。

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        min_price = float('inf')
        max_profit = 0
        
        for price in prices:
            if price < min_price:
                min_price = price
            else:
                max_profit = max(max_profit, price - min_price)
                
        return max_profit