leetcode -- 買賣股票的最佳時機
阿新 • • 發佈:2018-12-24
122
給定一個數組,它的第 i 個元素是一支給定股票第 i 天的價格。
設計一個演算法來計算你所能獲取的最大利潤。你可以儘可能地完成更多的交易(多次買賣一支股票)。
注意:你不能同時參與多筆交易(你必須在再次購買前出售掉之前的股票)。
示例 1:
輸入: [7,1,5,3,6,4]
輸出: 7
解釋: 在第 2 天(股票價格 = 1)的時候買入,在第 3 天(股票價格 = 5)的時候賣出, 這筆交易所能獲得利潤 = 5-1 = 4 。
隨後,在第 4 天(股票價格 = 3)的時候買入,在第 5 天(股票價格 = 6)的時候賣出, 這筆交易所能獲得利潤 = 6-3 = 3 。
示例 2:
輸入: [1,2,3,4,5]
輸出: 4
解釋: 在第 1 天(股票價格 = 1)的時候買入,在第 5 天 (股票價格 = 5)的時候賣出, 這筆交易所能獲得利潤 = 5-1 = 4 。
注意你不能在第 1 天和第 2 天接連購買股票,之後再將它們賣出。
因為這樣屬於同時參與了多筆交易,你必須在再次購買前出售掉之前的股票。
示例 3:
輸入: [7,6,4,3,1]
輸出: 0
解釋: 在這種情況下, 沒有交易完成, 所以最大利潤為 0。
class Solution(object):
def maxProfit(self, prices) :
"""
:type prices: List[int]
:rtype: int
"""
income, buy = 0, -1
lenth = len(prices)
for i in range(lenth):
if i < lenth-1 and prices[i+1] > prices[i] and buy == -1:
buy = prices[i]
elif i > 0 and prices[i] < prices[i-1] and buy != -1:
income = income + prices[i-1] - buy
buy = prices[i]
elif i == lenth - 1 and buy != -1:
income = income + prices[i] - buy
return income
邏輯略複雜了一點,不過能保證在最少的操作下達到目的,如果不需要控制運算元的話邏輯可以簡單點
class Solution(object):
def maxProfit(self, prices):
"""
:type prices: List[int]
:rtype: int
"""
income, profit = 0, 0
lenth = len(prices)
for i in range(lenth-1):
profit = prices[i+1] - prices[i]
if profit> 0:
income += profit
return income