【动态规划】 309 最佳买卖股票含冷冻期

题目

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

思路

股票问题不含冷冻期
dp【】【3】增加一个维度，是冷冻期的情况。冷冻期前一天只有一种情况，就是现金，也就是卖出
dp表示的是今天的操作带来的收益。
收益是+price，支出是—price

代码

public int maxProfit(int[] prices) {
    if(prices.length<2) return 0;
    int [][]dp  = new int[prices.length][3];
    //0 :现金(卖出) 1：股票（买入）  2:冷冻
    dp[0][0] = 0;
    dp[0][1] = -prices[0];
    dp[0][2] = 0;
    for(int i = 1;i<prices.length;i++){
    // 今天是现金：前一天是现金或者前一天买入今天卖出，然后是现金
        dp[i][0] = Math.max(dp[i-1][0],dp[i-1][1]+prices[i]);
        //买入的前一天是冷冻期或者股票
        //要买前一天只能是冷冻期或者买入状态
        // 今天是股票：前一天是股票或者前一天是冷冻然后今天买入股票
        dp[i][1] = Math.max(dp[i-1][2]-prices[i],dp[i-1][1]);
        //冷冻期前一天是卖出
        dp[i][2] = dp[i-1][0];
    }
    return Math.max(dp[prices.length - 1][0], dp[prices.length - 1][2]);

}