LeetCode 每日一题188. 买卖股票的最佳时机 IV

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

给定一个整数数组 prices ，它的第 i 个元素 prices[i] 是一支给定的股票在第 i 天的价格。

设计一个算法来计算你所能获取的最大利润。你最多可以完成 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 。

提示：

• 0 <= k <= 109
• 0 <= prices.length <= 1000
• 0 <= prices[i] <= 1000

方法一：动态规划

买卖股票系列最后一题。同样，用「动态规划」的思想来解决。

• 状态定义：buy[k] 表示每次买入股票状态的现金最大值，sell[k] 表示每次卖出股票状态的现金最大值

• 初始化：为了方便操作，可以把每一次买入股票状态都设为 负的第一天的股价

• 状态转移方程：

  1. 第 j 次买入状态，继续保持 或 用第 j - 1 次卖出的钱买入股票：buy[j] = Math.max(buy[j], sell[j - 1] - prices[i])
  2. 第 j 次卖出状态，继续保持 或 卖出第 j 次购买的股票：sell[j] = Math.max(sell[j], buy[j] + prices[i])
  3. 注意：第一次买入股票时是负的当日股价。

• 返回最后一次卖出股票的现金数 sell[k - 1]

public int maxProfit(int k, int[] prices) {
    int n = prices.length;
    if (n < 2 || k == 0) {
        return 0;
    }
    k = Math.min(k, n / 2);
    // 状态定义
    int[] buy = new int[k];
    int[] sell = new int[k];
    // 初始化
    for (int i = 0; i < k; i++) {
        buy[i] = -prices[0];
    }
    // 状态转移
    for (int i = 1; i < n; i++) {
        for (int j = 0; j < k; j++) {
            if (j == 0) {
            	// 第一次买入是负的当日股价
                buy[0] = Math.max(buy[0], -prices[i]);
            } else {
                buy[j] = Math.max(buy[j], sell[j - 1] - prices[i]);
            }
            sell[j] =  Math.max(sell[j], buy[j] + prices[i]);
        }
    }
    return sell[k - 1];
}

执行结果