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

shttps://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-with-cooldown
给定一个整数数组prices，其中第 prices[i] 表示第 i 天的股票价格 。​

设计一个算法计算出最大利润。在满足以下约束条件下，你可以尽可能地完成更多的交易（多次买卖一支股票）:

卖出股票后，你无法在第二天买入股票 (即冷冻期为 1 天)。
注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）

思路：
设dp[i] 为第i天结束后获得的最大收益，那么这个dp[i]有三种状态：
dp[i][0] : 目前手中持有一支股票
dp[i][1] : 目前手中不持有股票，且处于冷冻期，i+1天无法购买股票
dp[i][2] : 目前手中不持有股票，且不处于冷冻期
对于dp[i]的状态，我们应从dp[i - 1]中转移过来：

1. dp[i][0]的情况：目前手里持有的一支股票可以是从i-1天买进的，即dp[i - 1][0]，或者是第i天买进的，那么i-1就不是冷冻期，即dp[i - 1][2]，其最大收益还要减去第i天买进的钱，即dp[i - 1][2] - prices[i]，可得转移方程为：
   dp[i][0] = max(dp[i - 1][0], dp[i - 1][2] - prices[i])
2. dp[i][1]的情况：此种情况说明在i天的时候卖出了股票，即i-1天时持有股票dp[i - 1][0]，同时加上卖出股票获得的收益，即dp[i][1] = dp[i - 1][0] + prices[i]
3. dp[i][2]的情况：此种情况可能i - 1天为冷冻期，即dp[i - 1][1]，也可能i - 1天不是冷冻期也不持有股票，即dp[i - 1][2]，可得转移方程为：
   dp[i][2] = max(dp[i - 1][1], dp[i - 1][2])

在最后一天持有股票也是没有意义的，所以最终结果是max(dp[n - 1][1], dp[n - 1][2])

var maxProfit = function(prices) {
    let n = prices.length
    let dp = Array.from(Array(n), () => Array(4).fill(0));
    dp[0][0] = 0 - prices[0];

    for (let i = 1; i < n; i++) {
        dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][2] - prices[i]);
        dp[i][1] = dp[i - 1][0] + prices[i];
        dp[i][2] = Math.max(dp[i - 1][1], dp[i - 1][2]);
    }

    return Math.max(dp[n - 1][1], dp[n - 1][2])
};

状态压缩：

var maxProfit = function(prices) {
    const len = prices.length;

    let sell = 0;
    let freeze = 0;
    let buy = -prices[0];

    for(let i = 1; i < len; i++) {
        let temp = sell;
        sell = Math.max(sell, freeze);
        freeze = buy + prices[i];
        buy = Math.max(buy, temp - prices[i]);
    }

    return Math.max(sell, freeze);
};