代码随想录算法训练营第五十天 123.买卖股票的最佳时机III、 188.买卖股票的最佳时机IV

代码随想录算法训练营第五十天 | 123.买卖股票的最佳时机III、 188.买卖股票的最佳时机IV

123.买卖股票的最佳时机III

题目链接：123. 买卖股票的最佳时机 III - 力扣（LeetCode）

class Solution {
    public int maxProfit(int[] prices) {
        int len = prices.length;
        if(len <= 1) return 0;
        
        // 只要明确了dp数组及下标的含义就能很好的推导
        /*
         * 定义每一天的dp有 5 种状态:
         * 0: 没有操作, 1: 第一次持有, 2: 第一次不持有, 3: 第二次持有, 4: 第二次不持有
         * dp[i][x] 表示第 i 天 某种状态下的利润
         * 其实我们可以不设置，‘0. 没有操作’ 这个状态，因为没有操作，手上的现金自然就是0
         */
        int[][] dp = new int[len][5];
        dp[0][1] = -prices[0];
        dp[0][3] = -prices[0];
        
        for(int i = 1; i < len; ++i) {
            dp[i][1] = Math.max(dp[i-1][0] - prices[0], dp[i - 1][1]);
            dp[i][2] = Math.max(dp[i-1][1] + prices[0], dp[i - 1][2]);
            dp[i][3] = Math.max(dp[i-1][2] - prices[0], dp[i - 1][3]);
            dp[i][4] = Math.max(dp[i-1][3] + prices[0], dp[i - 1][4]);
        }
        
        return dp[len - 1][4];
    }
}

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

题目链接：188. 买卖股票的最佳时机 IV - 力扣（LeetCode）

class Solution {
    public int maxProfit(int[] prices) {
        int len = prices.length;
        if(len <= 1) return 0;
        
        int[][] dp = new int[len][2 * k + 1];
        for(int j = 1; j < 2 * k; j += 2){
            dp[0][j] = -prices[0];
        }
        
        for(int i = 1; i < len; ++i) {
            for(int j = 0; j < 2 * k - 1; j += 2) {
                dp[i][j+1] = Math.max(dp[i-1][j] - prices[i], dp[i-1][j+1]);
                dp[i][j+2] = Math.max(dp[i-1][j+1] + prices[i], dp[i-1][j+2]); 
            }
        }
        
        return dp[len - 1][2 * k];
    }
}

总结

就是在122. 买卖股票的最佳时机 II - 力扣（LeetCode）的基础上增加了更多状态

Ⅱ可以无限次持有， dp = new int[len] [2]， 0: 表示持有, 1: 表示不持有

Ⅲ只有两次持有机会，所以有五种状态 ：

​ 0: 没有操作, 1: 第一次持有, 2: 第一次不持有, 3: 第二次持有, 4: 第二次不持有

Ⅳ有k次机会，所以有2 * k + 1种状态：

​ 0: 没有操作, 1: 第一次持有, 2: 第一次不持有, 3: 第二次持有…

相比Ⅱ，Ⅲ和Ⅳ多了状态“0: 没有操作”，因为Ⅱ中没有次数限制，“1: 表示不持有”就包含了此状态，但是Ⅲ和Ⅳ强调持有的次数和先后顺序（比如说“2: 第一次不持有”一定是第一次持有再卖出了的状态），所以需增加一个初始状态“0: 没有操作”