动态规划|188. 买卖股票的最佳时机 IV

题目连接地址：188. 买卖股票的最佳时机 IV

状态转移方程：
$$dp[i][j][1]=max(dp[i-1][j][1],dp[i-1][j-1][0]-[i]);$$$$dp[i][j][0]=max(dp[i-1][j][0],dp[i-1][j][1]+[i]);$$
dp[i][j][1]代表第i天时，第j次买卖时持有股票时的收益；
dp[i][j][0]代表第i天时，第j次买卖时持有股票时的收益。

Java代码如下：

class Solution {
    public int maxProfit(int k, int[] prices) {
        //虽然我之前想到了使用三维dp来解决k次买卖的问题，但是对于
        //初始化和遍历顺序搞的不是很清楚。
        int n = prices.length;
        int[][][] dp = new int[n+1][k+1][2];

        for(int i = 0; i <= k; i++){
            dp[0][i][1] = -prices[0];
        }

        for(int i = 1; i <= n; i++){
            for(int j = 1; j <= k; j++){
                dp[i][j][1] = Math.max(dp[i-1][j][1],dp[i-1][j-1][0] - prices[i-1]);
                dp[i][j][0] = Math.max(dp[i-1][j][0],dp[i-1][j][1] + prices[i-1]);
            }
        }

        return dp[n][k][0];
        
        
    }
}