LeetCode—188. 买卖股票的最佳时机 IV(Best Time to Buy and Sell Stock IV)——分析及代码（Java）

一、题目

给定一个整数数组 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-iv
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

二、分析及代码

1. 动态规划

（1）思路

设 b[i][j] 表示第 i 天进行了第 j 笔交易（指完成出售）且当前持有股票时的最大利润，s[i][j] 表示第 i 天进行了第 j 笔交易且当前未持有股票时的最大利润，则状态转移方程为：

b[i][j] = max(b[i - 1][j], s[i - 1][j] - prices[i])
s[i][j] = max(s[i - 1][j], b[i - 1][j - 1] + prices[i])

可进一步简化为一维数组(s[i][j] 计算中使用的 b[i - 1][j - 1] 可用 b[i][j - 1] 代替，可直接带入证明）：

b[j] = max(b[j], s[j] - prices[i])
s[j] = max(s[j], b[j - 1] + prices[i])

迭代完成后，因为 s[j] 值一定不小于 b[j] 值，且实际获得最大利润不一定需要 k 次交易，因此在 s[j] 数组中选取最大值，即为所求最大利润。

（2）代码

class Solution {
    public int maxProfit(int k, int[] prices) {
        if (prices.length == 0)
            return 0;
        int n = Math.min(k, prices.length >> 1) + 1;//最多可交易次数(+1)
        int [] b = new int[n];//进行第j笔交易且当前持有股票
        int [] s = new int[n];//进行第j笔交易且当前未持有股票
        
        b[0] = -prices[0];
        s[0] = 0;
        for (int i = 1; i < n; i++)
            b[i] = s[i] = Integer.MIN_VALUE >> 1;//初始化为最小值的一半，避免溢出


        for (int i = 0; i < prices.length; i++) {
            b[0] = Math.max(b[0], s[0] - prices[i]);//s[0]始终为0
            for (int j = 1; j < n; j++) {
                b[j] = Math.max(b[j], s[j] - prices[i]);
                s[j] = Math.max(s[j], b[j - 1] + prices[i]);
            }
        }
        return Arrays.stream(s).max().getAsInt();//返回最后一天时，不超过给定交易数时最大利润
    }
}

（3）结果

执行用时 ：8 ms，在所有 Java 提交中击败了 52.90% 的用户；
内存消耗 ：36.2 MB，在所有 Java 提交中击败了 82.41% 的用户。

三、其他

暂无。