1. 程式人生 > >LeetCode演算法題122:買賣股票的最佳時機 II 解析

LeetCode演算法題122:買賣股票的最佳時機 II 解析

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

這個題用暴力法就有點太複雜了,得稍微轉換一下思路,最困擾我們的地方就是當1,2,3,4,5這型別的情況,我們不知道是需要賣出再買還是等最高價賣出,但是換個思路想一下,5-1等於(2-1)+(3-2)+(4-3)+(5-4),也就是說這種情況只要後面數字大於前面數字,那就疊加利潤,最後一定是最大利潤,如果中間有一個數是小的,那麼直接跳過即可,相當於分成了幾個這樣的數列。因此程式應該如下:

C++原始碼:

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int maxProfit = 0;
        for (int i=1;i<prices.size();i++)
            if (prices[i] > prices[i-1])
                maxProfit += prices[i] - prices[i-1];
        return maxProfit;
    }
};

python3

class Solution:
    def maxProfit(self, prices):
        """
        :type prices: List[int]
        :rtype: int
        """
        maxProfit = 0
        for i in range(1, len(prices)):
            if prices[i-1] < prices[i]:
                maxProfit += prices[i] - prices[i-1]
        return maxProfit