188. 买卖股票的佳时机 IV

题目

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

设计一个算法来计算你所能获取的最大利润。你最多可以完成 k 笔交易。

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

代码

class Solution {
    public int maxProfit(int k, int[] prices) {

        if (k>prices.length/2)
            return maxProfit_inf(prices);

        int n=prices.length;
        int[][][] dp=new int[n][k+1][2];
//        for (int i = 0; i < n; i++) {
//            dp[i][0][1]=Integer.MIN_VALUE;//没有操作次数时是不可能持有的
//        }

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

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

    }
     int maxProfit_inf(int[] prices) {
       
        int n =prices.length;
        if (n==0) return 0;
        int[][] dp=new int[n][2];
        for (int i = 0; i < n ; i++) {
            if (i==0){
                dp[i][0]=0;
                dp[i][1]=-prices[i];//base case
                continue;
            }
            dp[i][0]=Math.max(dp[i-1][0],dp[i-1][1]+prices[i]);
            dp[i][1]=Math.max(dp[i-1][1],dp[i-1][0]-prices[i]);

        }

        return dp[n-1][0];
    }
}

要点

• 股票买卖问题，使用有限状态机模板来进行解题
• 根据具体题目要求进行简化以及base 情况的修改即可
• dp表结构：