题:给定一个数组,它的第 i 个元素是一支给定股票第 i 天的价格。如果你最多只允许完成一笔交易(即买入和卖出一支股票),设计一个算法来计算你所能获取的最大利润。注意你不能在买入股票前卖出股票。
示例 :
输入: [7,1,5,3,6,4]
输出: 5
解释: 在第 2 天(股票价格 = 1)的时候买入,在第 5 天(股票价格 = 6)的时候卖出,最大利润 = 6-1 = 5 。注意利润不能是 7-1 = 6, 因为卖出价格需要大于买入价格。
个人思路:dp思想,将当前卖出所得利润与上一天卖出时的最佳利润进行比较取最大值。如果当前卖出所得利润低于从上一天买进时所得利润,直接重置,从上天开始买进股票,这点类似于最大子序和的思想。时间复杂度是O(n)。
class Solution:
def maxProfit(self, prices):
if len(prices)==0:
return 0
buy=prices[0]
sell=prices[0]
maxprofit=sell-buy
for i in range(1,len(prices)):
sell=prices[i]
end_node=sell-buy
if prices[i]-prices[i-1]>end_node:
buy=prices[i-1]
maxprofit=max(sell-buy,end_node,maxprofit)
return maxprofit
leecode的官方解法比较巧妙,将给定数组绘制成折线图,事实上我们求最大利润就是找到该折线中的谷和峰。
minprice=2^1000000
maxprice=0
for i in range(len(prices)):
if prices[i]<minprice:
minprice=prices[i]
elif prices[i]-minprice>maxprice:
maxprice=prices[i]-minprice
return maxprice