309.最佳买卖股票时机含冷冻期

Posted 2021-11-07 HardyDragon_CC

309.最佳买卖股票时机含冷冻期

思路

• 四种dp 状态：买入，卖出、今日卖出、冷冻
• 初始化只需要将买入的设置为价格负数，其他三个卖出利润都是0

class Solution {
    public int maxProfit(int[] prices) {
        int size = prices.length;
        if(size == 0) return 0;
        // 0 买入 1 卖出状态 2 今天卖出 3 冷冻期
        int[][] dp = new int[size][4];
        dp[0][0] = -prices[0];
        for(int i = 1;i < size;i++){
            // 可能是昨天买了、昨天是冷冻期，昨天是卖出状态
            dp[i][0] = Math.max(dp[i-1][0],Math.max(dp[i-1][1],dp[i-1][3])- prices[i]);
            // 前一天就是卖出、前一天是冷冻期
            dp[i][1] = Math.max(dp[i-1][1],dp[i-1][3]);
            // 昨天的买入状态 + 今天的股价
            dp[i][2] = dp[i-1][0] + prices[i];
            // 和昨天的卖出价格相同
            dp[i][3] = dp[i-1][2];
        }
        // 比较3种卖出的利润
        return Math.max(dp[size-1][1],Math.max(dp[size-1][2],dp[size-1][3]));
    }
}