LeetCode41动态规划part11● 123.买卖股票的最佳时机III ● 188.买卖股票的最佳时机IV

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

力扣题目链接

class Solution {
    public int maxProfit(int[] prices) {
        //dp[i][0] = 没买过
        //dp[i][1] = 第i天第一次持有手里的钱
        //dp[i][2] = 第i天第一次卖出去手里的钱（不一定是当天卖的，可能早就卖了
        //dp[i][3] = 第i天第二次持有手里的钱
        //dp[i][4] = 第i天第二次卖出去手里的钱（不一定是当天卖的，可能早就卖了
        //dp[i][1] = Math.max(dp[i - 1][1],-prices[i]);
        //dp[i][2] = Math.max(dp[i - 1][2],dp[i - 1][1] + prices[i]);
        //dp[i][3] = Math.max(dp[i - 1][3],dp[i - 1][2] - prices[i]);
        //dp[i][4] = Math.max(dp[i - 1][4],dp[i - 1][3] + prices[i]);
        //dp[0][1] = -prices[0];dp[0][3] = -prices[0]其他都是0
        //prices从前往后遍历

        int len = prices.length;
        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][1],dp[i - 1][0] - prices[i]);
        dp[i][2] = Math.max(dp[i - 1][2],dp[i - 1][1] + prices[i]);
        dp[i][3] = Math.max(dp[i - 1][3],dp[i - 1][2] - prices[i]);//dp[0][3]被初始化为0，这里会出问题
        dp[i][4] = Math.max(dp[i - 1][4],dp[i - 1][3] + prices[i]);
        }
        return dp[len - 1][2] > dp[len - 1][4] ? dp[len - 1][2] : dp [len - 1][4];
    }
}

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

力扣题目链接

class Solution {
    public int maxProfit(int k, int[] prices) {
        //dp[i][0] = 啥也没干
        //dp[i][1,3,5,^^^^ 2k - 1]已经买入
        //dp[i][2,4,6,^^^^ 2k]已经卖出

        //dp[0][1,3,^^^2k - 1] = - prices[0];
        if(k  == 0 )return 0;

        int len = prices.length;
        int[][] dp = new int[len][2 * k + 1];
        for(int i = 1;i < 2 * k;i+=2){
            dp[0][i] = -prices[0];
        }

        for(int i = 1;i < len;i++){
            for(int j = 1; j <= k;j++){
                dp[i][2 * j - 1] = Math.max(dp[i - 1][2 * j - 1],dp[i - 1][2 * j - 2] - prices[i]);//逗号后面是之前上一次卖出 - 这次买入
                dp[i][2*j] = Math.max(dp[i - 1][2 * j],dp[i - 1][2 * j - 1] + prices[i]);//逗号之后是之前上一次买入，这次卖出
            }
        }
        int max = 0;
        for(int i = 1;i <= k;i++){
            max = max > dp[len - 1][2 * i] ? max : dp[len - 1][2 * i];
        }
        return max;
    }
}