LeetCode 122. 買賣股票的最佳時機 II(C、C++、python)
阿新 • • 發佈:2018-12-15
給定一個數組,它的第 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。
C
int maxProfit(int* prices, int pricesSize) { int n=pricesSize; if(n==0 || n==1) { return 0; } int res=0; int ans=0; int cc; int temp=prices[0]; for(int i=1;i<n;i++) { cc=prices[i]-prices[i-1]; if(cc<0) { ans+=res; temp=prices[i]; res=0; } else { cc=prices[i]-temp; res=res>cc?res:cc; } } if(res>0) { ans+=res; } return ans; }
C++
class Solution { public: int maxProfit(vector<int>& prices) { int n=prices.size(); if(n==0 || n==1) { return 0; } int res=0; int ans=0; int temp=prices[0]; int cc; for(int i=1;i<n;i++) { cc=prices[i]-prices[i-1]; if(cc<0) { ans+=res; temp=prices[i]; res=0; } else { res=max(res,prices[i]-temp); } } if(res>0) { ans+=res; } return ans; } };
python
class Solution:
def maxProfit(self, prices):
"""
:type prices: List[int]
:rtype: int
"""
n=len(prices)
if n==0 or n==1:
return 0
res=0
ans=0
temp=prices[0]
for i in range(1,n):
cc=prices[i]-prices[i-1]
if cc<0:
ans += res
temp=prices[i]
res=0
else:
res=max(res,prices[i]-temp)
if res>0:
ans += res
return ans