1. 程式人生 > >122. 買賣股票的最佳時機 II(JavaScript)

122. 買賣股票的最佳時機 II(JavaScript)

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

思路:

相比於買賣股票的最佳時機 Ⅰ,這裡可以多次買賣股票,但有個限制,就是再次買股票時必須先將之前的股票賣掉。

設定兩個變數:buy和sell,分別代表買入的股票和賣出的股票,初始值均為prices[0]。

示例1 和示例2 告訴我們,對於一串序列,例如4個數,有兩種可能,買賣一次和買賣兩次,當第三個數大於等於第二個數時,買賣一次的利潤更高(或相等),當第三個數小於第二個數時,分兩次買賣能獲得更高的利潤。

因此,我們可以進行遍歷,若prices[i] >= sell,將sell設定為prices[i],否則,就把這次的買賣加入總利潤中,重新設定買賣金額為prices[i]。

遍歷完之後,最後一次的買賣尚未加入總利潤中,所以還得加一次。

/**
 * @param {number[]} prices
 * @return {number}
 */
var maxProfit = function(prices) {
  if (prices.length < 2) return 0;
  let buy = sell = prices[0];
  let max_profit = 0;
  for (let i in prices) {
    if (prices[i] >= sell) {
      sell = prices[i];
    } else {
      max_profit += sell - buy;
      buy = prices[i];
      sell = prices[i];
    }
  }
  max_profit += sell - buy;
  return max_profit;
};