给定一个数组,它的第 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。
偷懒想法:遍历两次
class Solution(object):
def maxProfit(self, prices):
"""
:type prices: List[int]
:rtype: int
"""
n = len(prices)
mp = 0
for i in range(n-1):
for j in range(i+1, n):
profit = prices[j]-prices[i]
if profit > 0:
mp = max(mp, profit)
return mp
结果发现:
于是遍历一次:用prices[0]为第一次的最小值,遍历取最大利润,如果有比prices[0]更小的直接替换,不用重新遍历。
class Solution(object):
def maxProfit(self, prices):
"""
:type prices: List[int]
:rtype: int
"""
if len(prices)<2:
return 0
max_profit = 0
min_price = prices[0]
for i in price:
min_price=min(min_price,i)
max_profit=max(max_profit, i-min_price)
return max_profit
优化下代码:
class Solution(object):
def maxProfit(self, prices):
"""
:type prices: List[int]
:rtype: int
"""
if prices==[]:
return 0
max_profit = 0
min_price = prices[0]
for i in prices[1:]:
if i < min_price:
min_price=i
max_profit=max(max_profit, i-min_price)
return max_profit