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

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

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

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

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

示例 1:

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

示例 2:

1
输入: prices = [1]

Solution

只需要考虑在持有股票时，买入的股票要是前2天的

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
class Solution {
    // 主方法，计算最大利润
    public int maxProfit(int[] prices) {
        int n = prices.length; // 获取数组的长度
        int[][] dp = new int[n + 2][2]; // 初始化二维数组 dp，大小为 (n+2) x 2

        // 初始化初始状态
        dp[0][0] = 0; // 第 0 天不持有股票的利润为 0
        dp[1][1] = Integer.MIN_VALUE / 2; // 第 1 天持有股票的利润设为一个极小值

        // 遍历每一天
        for (int i = 0; i < n; i++) {
            // 更新第 i + 2 天不持有股票的利润
            dp[i + 2][0] = Math.max(dp[i + 1][0], dp[i + 1][1] + prices[i]);
            // 更新第 i + 2 天持有股票的利润
            dp[i + 2][1] = Math.max(dp[i][0] - prices[i], dp[i + 1][1]);
        }

        // 返回第 n + 1 天不持有股票的最大利润
        return dp[n + 1][0];
    }
}