LeetCode 309 (中等，动态规划)

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

1.题目

给定一个整数数组，其中第 i 个元素代表了第 i 天的股票价格 。​。设计一个算法计算出最大利润。在满足以下约束条件下，你可以尽可能地完成更多的交易（多次买卖一支股票）:你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。。卖出股票后，你无法在第二天买入股票 (即冷冻期为 1 天)。。

示例1：
输入: [1,2,3,0,2]
输出: 3
解释: 对应的交易状态为: [买入, 卖出, 冷冻期, 买入, 卖出]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-with-cooldown
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2.题解

       本题不同的地方在于有一个冷却期，需要考虑的是如果我需要购买股票，需要前一天不能是卖出的，只能是前一天是冷却期，或者前一天什么都没干， 那么可以将问题转化一下，分为三种状态，dp[i][0], dp[i][1], dp[i][2], dp[i][0]表示今天不持有固表的最大值，或者今天是冷却期，dp[i][1]表示今天持有了股票的最大值，来源可能是昨天就持有的股票也可能是今天购买的股票，dp[i][2]表示今天卖出了股票的最大值。
        那么就可以进行判断各个情况是怎么得到的，如果是dp[i][0]，那么就有dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][2]); 表示昨天什么也没干，或者今天是冷却期的最大值，dp[i][1] = Math.max(dp[i - 1][1], dp[i - 1][0] - prices[i]);今天持有股票只能是昨天持有的和今天购买的最大值，而今天能购买的条件是昨天没有持有或者是冷却期，dp[i][2] = dp[i - 1][1] + prices[i];今天卖出，需要昨天持有的股票的最大值。

         dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][2]);
         dp[i][1] = Math.max(dp[i - 1][1], dp[i - 1][0] - prices[i]);
         dp[i][2] = dp[i - 1][1] + prices[i];

那么就可以初始化dp[0][1] = -1 * preices[0]表示第一天购买的情况，因为购买需要花钱，所以是负值。

• 代码如下

class Solution {
    public int maxProfit(int[] prices) {
        int n = prices.length;
        int[][] dp = new int[n][3];
        dp[0][1] = -1 * prices[0];
        for(int i = 1; i < n; i++){
            dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][2]);
            dp[i][1] = Math.max(dp[i - 1][1], dp[i - 1][0] - prices[i]);
            dp[i][2] = dp[i - 1][1] + prices[i];
        }
        return Math.max(dp[n - 1][0], dp[n - 1][2]);
    }
}