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

题目

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

思路

• dp[i][0]：第i天后手上持有股票的最大收益
  1. dp[i-1][0]手上持有股票的状态第i天不做任何操作之后还是处于手上持有股票状态
  2. dp[i-1][2]手上不持有股票，并且不处于冷冻期的状态第i天买入股票之后就处于手上持有股票状态
• dp[i][1]:第i天之后手上不持有股票，并且处于冷冻期的累计最大收益
  1. dp[i-1][0]手上持有股票第i天卖出股票之后就处于手上不持有股票，并且处于冷冻期的累计最大收益
• dp[i][2]:第i天之后手上不持有股票，并且不处于冷冻期的累计最大利益
  1. dp[i-1][1]手上不持有股票，并且处于冷冻期的状态不进行任何操作第i天结束后就是不持有股票并且不处于冷冻期
  2. dp[i-1][2]手上不持有股票，并且不处于冷冻期的状态不进行任何操作第i天结束后还是不持有股票并且不处于冷冻

代码

class Solution {
    public int maxProfit(int[] prices) {
        if(prices.length==0){
            return 0;
        }

        int n=prices.length;

        int[][] dp=new int[n][3];
        dp[0][0]=-prices[0];

        for(int i=1;i<n;i++){
            /*
            dp[i][0]：第i天后手上持有股票的最大收益
            1. dp[i-1][0]手上持有股票的状态第i天不做任何操作之后还是处于手上持有股票状态
            2. dp[i-1][2]手上不持有股票，并且不处于冷冻期的状态第i天买入股票之后就处于手上持有股票状态
             */
            
            dp[i][0]=Math.max(dp[i-1][0],dp[i-1][2]-prices[i]);

            /*
            dp[i][1]:第i天之后手上不持有股票，并且处于冷冻期的累计最大收益
            1. dp[i-1][0]手上持有股票第i天卖出股票之后就处于手上不持有股票，并且处于冷冻期的累计最大收益
             */
            
            dp[i][1]=dp[i-1][0]+prices[i];

            /* dp[i][2]:第i天之后手上不持有股票，并且不处于冷冻期的累计最大利益
            1. dp[i-1][1]手上不持有股票，并且处于冷冻期的状态不进行任何操作第i天结束后就是不持有股票并且不处于冷冻期          2. dp[i-1][2]手上不持有股票，并且不处于冷冻期的状态不进行任何操作第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]);
    }
}

复杂度

时间复杂度： O(n)
空间复杂度： O(n)