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

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

一、力扣123.买卖股票的最佳时机III

题目链接
思路：之前的类型是用数组记录当前是持有还是不持有，而这道题目多了两种状态，即为，第一次持有，第一次不持有，第二次持有，第二次不持有。这个第几次是状态。
第一次持有 dp[i][0]=目前是第一次持有的状态，可以是之前持有的，也可以是今天才持有的，反正是第一次持有，没有持有两次。
第一次不持有 dp[i][1]=可以是昨天就不是持有状态，也可以是今天才是非持有，也就是进天才卖掉。
第二次持有 dp[i][2]=可以昨天就是第二次持有，也可以是今天才买入才变成第二次持有。
第二次不持有 dp[i][3]=可以是昨天就不是第二次持有，也可以今天才第二次卖掉。

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

二、力扣188.买卖股票的最佳时机IV

题目链接
思路：和上面的类似，只不过是可以买入k次不是2次，那么dp数组变成dp[nums.length][2*k+1]即可

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