题目:
剑指 Offer 63. 股票的最大利润。
假设把某股票的价格按照时间先后顺序存储在数组中,请问买卖该股票一次可能获得的最大利润是多少?
链接:https://leetcode-cn.com/leetbook/read/illustrate-lcof/5w5qjr/
本题与主站 121 题相同:https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock/
示例:
输入: [7,1,5,3,6,4]
输出: 5
解释: 在第 2 天(股票价格 = 1)的时候买入,在第 5 天(股票价格 = 6)的时候卖出,最大利润 = 6-1 = 5 。
注意利润不能是 7-1 = 6, 因为卖出价格需要大于买入价格。
输入: [7,6,4,3,1]
输出: 0
解释: 在这种情况下, 没有交易完成, 所以最大利润为 0。
限制: 0 <= 数组长度 <= 10^5
思路:
设置两个变量profit和mark,分别用于存储利润和买入时间。
从数组的第一个元素开始遍历(起始为0),如果该元素小于数组的第0个元素,说明该点更适合买入,则买入节点更新为当前元素的索引;否则,计算利润,并与先前的利润值比较取最大。
最后返回profit值。
代码:
class Solution:
def maxProfit(self, prices: List[int]) -> int:
if len(prices) <= 1: return 0
profit, mark = 0, 0
for i, num in enumerate(prices[1:]):
if num < prices[mark]:
mark = i+1
else:
profit = max(profit, num-prices[mark])
return profit
时间复杂度O(n),空间复杂度O(1).