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

代码随想录

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

思路

思路：最多可以参与2笔交易，那么有三种情况：
不买卖、买卖一次、买卖两次

dp[i][j]表示第i天状态j的时候最大收益
j 有五种状态
0-无操作
1-第一次持有股票
2-第一次不持有股票
3-第二次持有股票
4-第二次不持有股票

递推公式：
1有两种可能，之前就持有股票，当天买入股票
dp[i][1]=max(dp[i-1][1],dp[i-1][0]-prices[i])
2有两种可能，之前就不持有，当天卖出
dp[i][2]=max(dp[i-1][2],dp[i-1][1]+prices[i])
3有两种可能，之前就持有，当天买入
dp[i][3]=max(dp[i-1][3],dp[i-1][2]-prices[i])
4有两种，之前就不持有，当天卖出
dp[i][4]=max(dp[i-1][4],dp[i-1][3]+prices[i])

初始化：
dp[0][0]=0第0天没有操作，
dp[0][1]=dp[0][3]=-prices[0]
dp[0][2]=dp[0][4]=0

dp[len][4]是最大的结果，因为如果是dp[len][2]可以当天买入再卖出

代码

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][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];

    }
}

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

思路

思路：跟上一题类似，分为2k+1种状态
状态0表示没有操作
状态1表示第1次买入
状态2表示第1次卖出
…
状态2k-1表示第k次买入
状态2k表示第k次卖出
第i天在 在每种状态的递推公式如下
dp[i][2j-1]=Math.max(dp[i-1][2j-1],dp[i-1][2j-2]-prices[i]);
dp[i][2j]=Math.max(dp[i-1][2j],dp[i-1][2j-1]+prices[i]);
分别是如果持有股票，那么从本来就持有和当天买入取最大
如果不持有，那么从本来就不持有和当天卖出取最大。

代码

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

        for(int i = 1;i<prices.length;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]);
            }
         
        }
        return dp[prices.length-1][2*k];

    }
}