1. FlyPython首页
  2. 数据结构与算法
  3. leetcode题解

「LeetCode每日一题」——121. 买卖股票的最佳时机

121. 买卖股票的最佳时机

链接:https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock/
难度:简单

题目

给定一个数组,它的第 i 个元素是一支给定股票第 i 天的价格。

如果你最多只允许完成一笔交易(即买入和卖出一支股票),设计一个算法来计算你所能获取的最大利润。

注意你不能在买入股票前卖出股票。

示例 1:

输入: [7,1,5,3,6,4]
输出: 5
解释: 在第 2 天(股票价格 = 1)的时候买入,在第 5 天(股票价格 = 6)的时候卖出,最大利润 = 6-1 = 5 。
注意利润不能是 7-1 = 6, 因为卖出价格需要大于买入价格。
示例 2:

输入: [7,6,4,3,1]
输出: 0
解释: 在这种情况下, 没有交易完成, 所以最大利润为 0。

思路

今天这个题,看着又像动态规划的题。

如果我们先不考虑动态规划,先试试枚举所有的情况,可能会超时,因为时间复杂度为O(n^2),那如果这里我们只遍历一遍呢。如下图所示,我们遍历的时候带着最低点和最大差值往前走,遍历完了,最大的利润也有了。

为啥这种思路会起作用呢,这是因为这思路就是来自于动态规划。这里就不详细解释动态规划的解法了,后期在面试真题解析中会有这种题型的总结。代码见方案代码。

方案代码

解决方法:

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        inf = int(1e9)
        maxprofit = 0
        minprice = inf
        for price in prices:
            maxprofit = max(maxprofit, price - minprice)
            minprice = min(price, minprice)
        return maxprofit

原创文章,作者:flypython,如若转载,请注明出处:http://flypython.com/algorithm/leetcode/213.html