LeetCode309最佳买卖股票时机含冷冻期(状态机dp)

题目链接

我们目前持有一支股票，对应的「累计最大收益」记为 f[i][0]；
我们目前不持有任何股票，并且处于冷冻期中，对应的「累计最大收益」记为 f[i][1]；
我们目前不持有任何股票，并且不处于冷冻期中，对应的「累计最大收益」记为 f[i][2]。
这里的「处于冷冻期」指的是在第 ii 天结束之后的状态。也就是说：如果第 i 天结束之后处于冷冻期，那么第 i+1 天无法买入股票。

    public int maxProfit(int[] prices) {
        if (prices.length == 0) return 0;
        int f[][] = new int[prices.length][3];
        //f[i][0] 手上持有股票的最大收益
        //f[i][1] 手上不持有股票 并且处于冷冻期的最大收益
        //f[i][2] 手上不持有股票 并且不处于冷冻期的最大收益
        f[0][0] = -prices[0];
        for (int i = 1; i < prices.length; i++) {
            f[i][0] = Math.max(f[i-1][0],f[i-1][2]-prices[i]);
            f[i][1] = f[i-1][0]+prices[i];
            f[i][2] = Math.max(f[i-1][1],f[i-1][2]);
        }
        return Math.max(f[prices.length-1][1],f[prices.length-1][2]);
    }