1. 程式人生 > >LeetCode 122. 買賣股票的最佳時機 II(C、C++、python)

LeetCode 122. 買賣股票的最佳時機 II(C、C++、python)

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