动态规划——最佳买卖股票时机含冷冻期

LeetCode（309）：最佳买卖股票时机含冷冻期

题目描述

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

解题思路

1、动态规划
与分治算法思想一致，都是将复杂问题分解成若干小问题，最后通过小问题进行求。不同的是，分治算法各个小问题之间是相互独立的；而动态规划算法中各个小问题之间相互影响。
2、题解
定义：若买入股票则总金额-prices[i];若卖出股票则总金额+prices[i]。
第i天可以处于三种状态中的一种：
0：第i天持有股票。
1：第i天卖出股票，即第i+1天不能购买股票。
2：第i天未持有股票。
创建数组profit[prices.length][3],用来存储某天处于某种状态下的最大盈利。
若第i天处于0：代表目前持有股票，则profit[i][0] = Math.max(profit[i-1][0],profit[i-1][2]-prices[i]);
若第i天处于1：代表今天抛出股票，则profitp[i][1]=profit[i-1][0]+prices[i];
若第i天处于2：代表今天未持有股票，则profit[i][2]=Math.max(profit[i-1][1],profit[i-1][2]);
初始值：profit[0][0] = - prices[0];profit[0][1]=0;profit[0][2]=0;

代码实现

public int maxProfit(int[] prices) {
        //考虑特殊情况
        if(prices.length < 2){
            return 0;
        }

        //创建数组存放对应状态下的最大盈利
        int[][] profit = new int[prices.length][3];
        
        //初始化
        profit[0][0] = -prices[0];
        profit[0][1] = 0;
        profit[0][2] = 0;
        
        //循环从第1天开始
        for(int i = 1;i < prices.length;i++){
            profit[i][0] = Math.max(profit[i - 1][0],profit[i-1][2]-prices[i]);
            profit[i][1] = profit[i - 1][0] + prices[i];
            profit[i][2] = Math.max(profit[i-1][2],profit[i-1][1]);
          }
          //返回最终结果，最后一天不能有剩余股票未抛出
          return Math.max(profit[prices.length - 1][1],profit[prices.length - 1][2]);

    }

执行效率