LeetCode Q123 买卖股票的最佳时机 3

思路

统一为三维数组的dp问题。dp[n][2][k] ,n为天数，第二维表示当天是否买入股票，k表示限制买卖的次数。

每天的利润则取决于当天的操作。只有两种选择，操作或者不操作。不操作则直接和前一天利润相同。操作 则按操作的不同增减利润

代码

class Solution {
    public int maxProfit(int[] prices) {
        if(prices.length==0) return 0;
       int[][][] dp=new int[prices.length][2][3];
       dp[0][0][0]=dp[0][0][1]=dp[0][0][2]=0;//第一天啥都没做的情况下利润为0
       dp[0][1][0]=dp[0][1][1]=dp[0][1][2]=-prices[0];//第一天持有股票，虽然后面两种情况不存在但还是初始化为-的第一天股票
       for(int i=1;i<prices.length;i++){
           dp[i][0][0]=0; //当天没有仅限交易，   且之前也没有进行交易，则表示截止丹田利润一直为0
           dp[i][0][1]=Math.max(dp[i-1][0][1],dp[i-1][1][0]+prices[i]); //两种选择，前天没有动作或者进行了卖出，因为当天的状态是持有股票的,因此选择两种中的利润较大者
           dp[i][0][2]=Math.max(dp[i-1][0][2],dp[i-1][1][1]+prices[i]); //同上
           dp[i][1][0]=Math.max(dp[i-1][1][0],dp[i-1][0][0]-prices[i]); //没有持有的情况下进行了持有，那么利润就要降低
           dp[i][1][1]=Math.max(dp[i-1][1][1],dp[i-1][0][1]-prices[i]);
           dp[i][1][2]=0; //手里还有股票，且已经不能再进行交易，这种情况肯定要避免，肯定不是最优解，直接为0
       }
       return dp[prices.length-1][0][2];
    }
}