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

class Solution {
    public int maxProfit(int[] prices) {
        int n = prices.length;
        //dp[i][j][k]表示在第i天交易了j次后在k状态下的最大利润
        int[][][] dp = new int[n][3][2];
        //边界条件
        dp[0][0][0] = 0;//第0天交易0次未持有的状态下收益为0
        dp[0][0][1] = -prices[0];//以此类推
        dp[0][1][0] = 0;
        dp[0][1][1] = -prices[0];
        dp[0][2][0] = 0;
        dp[0][2][1] = -prices[0];
        for(int i = 1; i < n; i++){
            dp[i][0][0] = dp[i - 1][0][0];
            dp[i][0][1] = Math.max(dp[i - 1][0][0] - prices[i],dp[i - 1][0][1]);
            dp[i][1][0] = Math.max(dp[i - 1][0][1] + prices[i],dp[i - 1][1][0]);
            dp[i][1][1] = Math.max(dp[i - 1][1][0] - prices[i],dp[i - 1][1][1]);
            dp[i][2][0] = Math.max(dp[i - 1][1][1] + prices[i],dp[i - 1][2][0]);  
        }
        int a = Math.max(dp[n - 1][0][0], dp[n - 1][0][1]);
        int b = Math.max(dp[n - 1][1][0], dp[n - 1][1][1]);
        return Math.max(Math.max(a,b) , dp[n - 1][2][0]);
    }
}