leetcode 刷题day40动态规划Part09 买卖股票（188.买卖股票的最佳时机IV、309.最佳买卖股票时机含冷冻期、714.买卖股票的最佳时机含手续费）

188.买卖股票的最佳时机IV

思路：III是最多可以买卖2次股票，IV是最多买卖K次。

前一题我们定义dp数组时设置了5个状态，同理，这个题目设置2*K+1个状态。0状态代表没有任何操作的状态，奇数表示买入，偶数为卖出。递归公式和前一个题目一致。

代码如下：

class Solution {
    public int maxProfit(int k, int[] prices) {
        int len=prices.length;
        if(len==0) return 0;
        int[][] dp=new int[len][2*k+1];
        for(int i=1;i<2*k;i+=2){
            dp[0][i]=0-prices[0];
        }
        for(int i=1;i<len;i++){
            for(int j=0;j<2*k-1;j+=2){
                dp[i][j+1]=Math.max(dp[i-1][j+1],dp[i-1][j]-prices[i]);
                dp[i][j+2]=Math.max(dp[i-1][j+2],dp[i-1][j+1]+prices[i]);
            }
        }
        return dp[len-1][2*k];
    }
}

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

思路：本题加了一个冷冻期，状态更多了。

1、确定dp数组以及下标的含义
具体可以区分出如下四个状态：
0-状态一：持有股票状态（今天买入股票，或者之前就买入了股票一直持有）
不持有股票状态，这里就有两种卖出股票状态：
- 1-状态二：保持卖出股票的状态（两天前就卖出了股票，度过一天冷冻期。或者是前一天就是卖出股票状态，一直没操作）
- 2-状态三：今天卖出股票
3-状态四：今天为冷冻期状态，但冷冻期状态不可持续，只有一天

2、递推公式：
1）达到买入股票状态（状态一）即：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]);

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

• 操作一：前一天就是状态二
• 操作二：前一天是冷冻期（状态四）
  dp[i][1] = max(dp[i - 1][1], dp[i - 1][3]);

3）达到今天就卖出股票状态（状态三），即：dp[i][2] ，只有一个操作：昨天一定是持有股票状态（状态一），今天卖出。即：dp[i][2] = dp[i - 1][0] + prices[i];

4）达到冷冻期状态（状态四），即：dp[i][3]，只有一个操作：昨天卖出了股票（状态三）dp[i][3] = dp[i - 1][2];

3、dp数组如何初始化
这里主要讨论一下第0天如何初始化。
1）如果是持有股票状态（状态一）那么：dp[0][0] = -prices[0]，一定是当天买入股票。
2）保持卖出股票状态（状态二），从 「状态二」的定义看很难明确应该初始多少，这种情况看递推公式。如果i为1，第1天买入股票，那么递归公式中需要计算 dp[0][1] - prices[1]，dp[0][1] （即第0天的状态二）只能初始为0。
3）今天卖出了股票（状态三），同上分析，dp[0][2]初始化为0，dp[0][3]也初始为0。

最后结果是取 状态二，状态三，和状态四的最大值，最后一天如果是冷冻期也可能是最大值。

代码如下：

class Solution {
    public int maxProfit(int[] prices) {
        int len=prices.length;
        if(len<=1) return 0;
        int[][] dp=new int[len][4];
        dp[0][0]=0-prices[0];
        for(int i=1;i<len;i++){
            dp[i][0]=Math.max(dp[i-1][0],Math.max(dp[i-1][1]-prices[i],dp[i-1][3]-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];
        }
        return Math.max(dp[len-1][1],Math.max(dp[len-1][2],dp[len-1][3]));
    }
}

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

思路：每买卖一次需要提交手续费，可以认为每次卖出的时候支付手续费。
也就是在之前的递归公式卖出状态时减去手续费。

代码如下：

class Solution {
    public int maxProfit(int[] prices, int fee) {
        int len=prices.length;
        if(prices==null || len==0) return 0;
        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 dp[len-1][1];
    }
}