LeeCode 121. 買賣股票的最佳時機(自己的算法)

題:給定一個數組,它的第 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
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章