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

一、题目描述

给定一个整数数组prices，其中第 prices[i] 表示第 i 天的股票价格 。​

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

• 卖出股票后，你无法在第二天买入股票 (即冷冻期为 1 天)。

注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。

示例 1：

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

示例 2：

输入: prices = [1]
输出: 0

二、解题

动态规划

dp[i]表示第 i 天结束之后的累计最大收益，不过这个收益包含了几种情况，分为冷静期，当卖出之后的一天不能买入。所以需要在增加一维：

• dp[i][0]: 手里有一支股票，并且对应着最大的收益
• dp[i][1]: 手里没有股票，并且处于冷冻期，不能买入卖出
• dp[i][2]: 手里没有股票 但是不处于冷冻期，对应有最大的收益

1、状态转移方程初始化:

第 0 天时，如果持有股票，那么只能是在第 0 天买入的，对应负收益 −prices[0]；
如果不持有股票，那么收益为0。注意到第 0 天实际上是不存在处于冷冻期的情况的，但我们仍然可以将对应的状态 f[0][1] , f[0][1] 置为零。

dp[0][0] = -prices[0];
dp[0][1] = 0;
dp[0][1] = 0;

2、状态转移方程:

1. 对于dp[i][0],该股票可以在第i-1天持有，对应的状态转移方程为dp[i][0] = dp[i-1][0];或者在第i天买入，那么第i-1天就不能持有股票且不在冷冻期。dp[i][0] = 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] = dp[i−1][0]+prices[i];
   dp[i][1] = dp[i−1][0]+prices[i]
3. 对于dp[i][2],在第 i 天结束之后不持有任何股票并且不处于冷冻期，说明当天没有进行任何操作，即第 i-1 天时不持有任何股票：如果处于冷冻期 dp[i][2] = dp[i−1][1];如果不处于冷冻期，对应的状态为dp[i][2] = dp[i-1][2]。
   dp[i][2] = max{dp[i−1][1]，dp[i-1][2]}；

class Solution {
    public int maxProfit(int[] prices) {
        if (prices.length == 0) {
            return 0;
        }
        int length = prices.length;
        //动态规划，分为三种情况 dp[i]第i天结束后累计的最大收益
        // dp[i][0]: 手里有一支股票，并且对应着最大的收益
        // dp[i][1]: 手里没有股票，并且处于冷冻期，不能买入卖出
        // dp[i][2]: 手里没有股票 但是不处于冷冻期，对应有最大的收益
        int[][] dp = new int[length][3];
        dp[0][0] = -prices[0];
        for(int i = 1;i<length;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[length-1][1],dp[length-1][2]);
    }
}