代码随想录day53

前言

今天又是卖股票专场，之前是买一次，买无数次，今天变成了2次，K次

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

思路：之前是两个状态：就是持有股票和不持有股票
现在限定成两次，那么我们就变成了五种状态
1.无操作
2.第一次持有股票
3.第一次不持有股票
4.第二次持有股票
5.第二次不持有股票

class Solution {
    public int maxProfit(int[] prices) {
        // 这个时候我们可以定义五种状态
        // dp[i][0] = dp[i - 1][0] 表示延续第i-1天的操作：也就是不操作
        // dp[i][1] :表示第一次持有股票 dp[i][1] = Math.max(dp[i-1][1],dp[i-1][0] - prices[i])
        // dp[i][2]:表示第一次不持有股票 dp[i][1] = Math.max(dp[i-1][2],dp[i-1][1] + prices[i]);
        // dp[i][3]:表示第二次持有股票 dp[i][3] = Math.max(dp[i-1][3],dp[i-1][1] - prices[i]);
        // dp[i[4]                      dp[i][4] = Math.max(dp[i-1][4],dp[i-1][3] + prices[i]);
        int[][] dp = new int[prices.length][5];
        dp[0][0] = 0;  //不操作
        dp[0][1] = -prices[0];
        dp[0][2] = 0;
        dp[0][3] = - prices[0];
        dp[0][4] = 0;

        for(int i = 1;i < prices.length;i++){
            dp[i][0] = dp[i-1][0];
            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[i][4] = Math.max(dp[i-1][4],dp[i-1][3] + prices[i]);

        }
        // return Math.max(dp[prices.length -1][4],dp[prices.length -1][2]);
        return dp[prices.length - 1][4];

    }
}

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

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

        // 第i天k次买入持有的最大利润
        int K = 2 * k;
        int[][] dp = new int[prices.length][2*k+1];
        for(int i = 0;i <= 2*k;i++){
            if(i % 2 == 1) dp[0][i] = - prices[0];
            else dp[0][i] = 0; 
        }
        // dp[i][j] 持有的最大利润
        // 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])
        for(int i = 1;i < prices.length;i++){
           
            for(int j = 0;j < 2*k-1;j += 2){
                    // 表示持有
                    dp[i][j+1] = Math.max(dp[i-1][j + 1], dp[i-1][j]-prices[i]);
                    // 表示不持有
                    dp[i][j+2] = Math.max(dp[i-1][j + 2],dp[i-1][j+1] + prices[i]);
                
            }
        }
        return dp[prices.length - 1] [2*k];


    }
}