代码随想录算法训练营Day51动态规划：309.最佳买卖股票时机含冷冻期，714.买卖股票的最佳时机含手续费

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

文章链接：代码随想录 (programmercarl.com)

思路：无思路

看完文章后的反思：（思路均摘自代码随想录）

（1）确定dp数组含义

dp[i][j]，第i天状态为j，所剩的最多现金为dp[i][j]

具体可以区分出如下四个状态：

• 状态一：持有股票状态（今天买入股票，或者是之前就买入了股票然后没有操作，一直持有）
• 不持有股票状态，这里就有两种卖出股票状态
  • 状态二：保持卖出股票的状态（两天前就卖出了股票，度过一天冷冻期。或者是前一天就是卖出股票状态，一直没操作）
  • 状态三：今天卖出股票
• 状态四：今天为冷冻期状态，但冷冻期状态不可持续，只有一天！

（2）确定递推公式

达到买入股票状态（状态一）即：dp[i][0]，有两个具体操作：

• 操作一：前一天就是持有股票状态（状态一），dp[i][0] = dp[i - 1][0]
• 操作二：今天买入了，有两种情况
  • 前一天是冷冻期（状态四），dp[i - 1][3] - prices[i]
  • 前一天是保持卖出股票的状态（状态二），dp[i - 1][1] - prices[i]

那么dp[i][0] = max(dp[i - 1][0], dp[i - 1][3] - prices[i], dp[i - 1][1] - prices[i]);

达到保持卖出股票状态（状态二）即：dp[i][1]，有两个具体操作：

• 操作一：前一天就是状态二
• 操作二：前一天是冷冻期（状态四）

dp[i][1] = max(dp[i - 1][1], dp[i - 1][3]);

达到今天就卖出股票状态（状态三），即：dp[i][2] ，只有一个操作：

昨天一定是持有股票状态（状态一），今天卖出

即：dp[i][2] = dp[i - 1][0] + prices[i];

达到冷冻期状态（状态四），即：dp[i][3]，只有一个操作：

昨天卖出了股票（状态三）

dp[i][3] = dp[i - 1][2];

（3）初始化

dp[0][0] = -prices[0]，dp[0][1] = 0，dp[0][2] = 0，dp[0][3] = 0。

（4）确定遍历顺序

从前往后

（5）举例推导

代码实现遇到的问题：

dp[i][0]和dp[i][1]的递推公式推导出现问题

Java代码：

class Solution {
    public int maxProfit(int[] prices) {
        //判断特殊情况
        if(prices.length == 0 || prices == null){
            return 0;
        }
        if(prices.length == 1){
            return 0;
        }
        int len = prices.length;
        /*定义四种状态
        0：持有股票状态（今天买入股票，或者是之前就买入了股票然后没有操作，一直持有）
        1：保持卖出股票的状态（两天前就卖出了股票，度过一天冷冻期。或者是前一天就是卖出股票状态，一直没操作）
        2：今天卖出股票
        3：今天为冷冻期状态，但冷冻期状态不可持续，只有一天！
        */
        //(1)dp数组含义，dp[i][j]表示在第i天j状态下所剩余的最大金额
        int[][] dp = new int[len][4];
        //(2)确定递推公式
        /*
        dp[i][0] = Math.max(dp[i - 1][0],dp[i - 1][3] - prices[i],dp[i - 1][1] - prices[i]);
        dp[i][1] = Math.max(dp[i - 1][1],dp[i - 1][3]);
        dp[i][2] = dp[i - 1][0] + prices[i];
        dp[i][3] = dp[i - 1][2];
        */
        //(3)初始化
        dp[0][0] = -prices[0];
        dp[0][1] = 0;
        dp[0][2] = 0;
        dp[0][3] = 0;
        //(4)确定遍历顺序，从前往后
        for(int i = 1; i < len;i++){
            int dpi0 = Math.max(dp[i - 1][0],dp[i - 1][3] - prices[i]);
            dp[i][0] = Math.max(dpi0,dp[i - 1][1] - prices[i]);
            dp[i][1] = Math.max(dp[i - 1][1],dp[i - 1][3]);
            dp[i][2] = dp[i - 1][0] + prices[i];
            dp[i][3] = dp[i - 1][2];
        }
        int res0 = Math.max(dp[len - 1][0],dp[len - 1][1]);
        int res1 = Math.max(dp[len - 1][2],dp[len - 1][3]);
        return Math.max(res0,res1);
    }
}

714.买卖股票的最佳时机含手续费

文章链接：代码随想录 (programmercarl.com)

思路：只是多了一个支付手续费的过程，在卖出股票时减去相应的手续费即可，动规五步曲

（1）确定dp数组含义，dp[i][j]表示第i天在j状态下最多剩余的金额，有0，1两个状态

0状态：买入股票（持有股票）

1状态：卖出股票（不持有股票）

（2）确定递推公式

dp[i][0]有两种情况：(a) 前一天就买入股票了，今天保持原状 (b) 今天刚买入股票

因此，dp[i][0] = Math.max(dp[i - 1][0],dp[i - 1][1] - prices[i]);

dp[i][1]也有两种情况：(a) 前一天就卖了股票，今天保持原状 (b) 今天刚卖，说明之前买入过股票

因此,dp[i][1] = Math.max(dp[i - 1][1] - fee,dp[i - 1][0] + prices[i] - fee);

（3）初始化

dp[0][0] = -prices[0];

dp[0][1] = 0;

（4）从前向后遍历

（5）举例推导

代码实现遇到的问题：

（1）dp[i][1]中的dp[i - 1][1] - fee写错，dp[i - 1][1]表示在前一天下的卖出股票了，今天保持原状，说明手续费fee已经在前一天给了，因此无需减去fee

Java代码：（思路如上）

class Solution {
    public int maxProfit(int[] prices, int fee) {
        //先判断特殊情况
        if(prices.length == 1){
            return 0;
        }
        int len = prices.length;
        int[][] dp = new int[len][2];
        dp[0][0] = -prices[0];
        dp[0][1] = 0;
        for(int i = 1; i < len;i++){
            dp[i][0] = Math.max(dp[i - 1][0],dp[i - 1][1] - prices[i]);
            dp[i][1] = Math.max(dp[i - 1][1],dp[i - 1][0] + prices[i] - fee);
        }
        return Math.max(dp[len - 1][0],dp[len - 1][1]);
    }
}