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

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

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

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

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

示例 1:

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

示例 2:

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

提示:

  • 1 <= prices.length <= 5000
  • 0 <= prices[i] <= 1000
    public int maxProfit(int[] prices) {
        // 创建二维数组,用来表示三种状态
        int[][] dp = new int[prices.length][3];
        //[0][0] 不持有 [0][1] 持有 [0][2] 冷冻期
        // 第一天如果持有股票,利润肯定是 -prices[0]
        dp[0][1] = -prices[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][2] - prices[i]
            dp[i][1] = Math.max(dp[i - 1][1], dp[i - 1][2] - prices[i]);
            // 冷冻期递推公式: 前一天不持有
            dp[i][2] = dp[i - 1][0];
        }
        // 最后返回不持有状态即可
        return dp[prices.length - 1][0];
    }

你可能感兴趣的:(算法,数据结构)