188. 买卖股票的最佳时机 IV #

原题链接：LeetCode 188. 买卖股票的最佳时机 IV

题目描述 #

给你一个整数数组 prices 和一个整数 k ，其中 prices[i] 是某支给定的股票在第 i天的价格。

设计一个算法来计算你所能获取的最大利润。你最多可以完成 k 笔交易。也就是说，你最多可以买 k 次，卖 k 次。

**注意：**你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。

示例 1：

• 输入： k = 2, prices = [2,4,1]
• 输出： 2
• 解释： 在第 1 天 (股票价格 = 2) 的时候买入，在第 2 天 (股票价格 = 4) 的时候卖出，这笔交易所能获得利润 = 4-2 = 2 。

示例 2：

• 输入： k = 2, prices = [3,2,6,5,0,3]
• 输出： 7
• 解释： 在第 2 天 (股票价格 = 2) 的时候买入，在第 3 天 (股票价格 = 6) 的时候卖出, 这笔交易所能获得利润 = 6-2 = 4 。 随后，在第 5 天 (股票价格 = 0) 的时候买入，在第 6 天 (股票价格 = 3) 的时候卖出, 这笔交易所能获得利润 = 3-0 = 3 。

提示：

• 1 <= k <= 100

• 1 <= prices.length <= 1000

• 0 <= prices[i] <= 1000

难度： Hard

题解代码 #

/**
 * @param {number} k
 * @param {number[]} prices
 * @return {number}
 */
var maxProfit = function(k, prices) {
  const profit = Array(k)
  for (let i = 0; i <= k; i++) {
    profit[i] = {
      buy: -prices[0],
      sell: 0
    }
  }

  for (let i = 0; i < prices.length; i++) {
    for (let j = 1; j <= k; j++) {
      profit[j] = {
        buy: Math.max(profit[j].buy, profit[j - 1].sell - prices[i]),
        sell: Math.max(profit[j].sell, profit[j].buy + prices[i])
      }
    }
  }
  return profit[k].sell
};