題:給定一個數組,它的第 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