代码随想录第四十二天|买卖股票的最佳时机 III|买卖股票的最佳时机 IV

买卖股票的最佳时机 III

这题不能无限次买卖了，只能有两次买卖，对于这种只有有限次事件的问题，可以把所有可能情况的递推公式都列出来，再进行遍历。

本题就分为五种情况，dp[i][0]表示在第i天不持有股票状态的最大利润为dp[i][0]，

dp[i][1]表示在第i天第1次不持有股票状态的最大利润为dp[i][1].

dp[i][2]表示在第i天第1次持有股票状态的最大利润为dp[i][2].

dp[i][3]表示在第i天第2次不持有股票状态的最大利润为dp[i][3].

dp[i][4]表示在第i天第2次持有股票状态的最大利润为dp[i][4].

递推公式：dp[i][0]=dp[i-1][0]；

dp[i][1]=max(dp[i-1][1],dp[i-1][0]-price[i]);

dp[i][2]=max(dp[i-1][2],dp[i-1][1]+price[i]);

dp[i][3]=max(dp[i-1][3],dp[i-1][2]-price[i]);

dp[i][4]=max(dp[i-1][4],dp[i-1][3]+price[i]);

初始化：dp[0][0]=0，dp[0][1]=0；dp[0][2]=-price[0];dp[0][3]=0;dp[0][4]=-price[0];

遍历顺序，从前往后

class Solution {
    public int maxProfit(int[] prices) {
        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 dp[prices.length-1][4];
    }
}

买卖股票的最佳时机 IV

这是将上一题的2次买卖扩展成k次，解题方法一样，将所有情况列出，只不过需要使用循环来进行初始化赋值和创建递推公式

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