算法打卡：第九章 动态规划part09

今日收获：买卖股票的最佳时机Ⅳ，最佳买卖股票时机含冷冻期，买卖股票的最佳时机含手续费，股票总结

1. 买卖股票的最佳时机Ⅳ

题目链接：188. 买卖股票的最佳时机 IV - 力扣（LeetCode）

思路：需要加定义2*k+1个状态，奇数代表持有股票，偶数代表不持有股票；持有股票可能是前一天持有股票或者前一天没有股票但今天卖出了，不持有股票可能是前一天不持有股票或者前一天持有股票但是今天卖出了。

方法：

class Solution {
    public int maxProfit(int k, int[] prices) {
        int len=prices.length;

        int[][] dp=new int[len][2*k+1];

        // 偶数不持有，奇数持有
        for (int i=1;i<=2*k-1;i+=2){
            dp[0][i]=-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];
    }
}

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

题目链接：309. 买卖股票的最佳时机含冷冻期 - 力扣（LeetCode）

思路：将dp数组的状态设置为4种，0为持有股票，1为卖出股票，2为保持卖出的状态（冷冻期后），3为冷冻期。

（1）0持有股票可以前一天就有，也可以前一天是2或3状态，今天买入股票

（2）1卖出股票必须是前一天持有股票

（3）2状态可以是前一天就是保持卖出状态2，或者前一天是冷冻期

（4）冷冻期的前一天必须是卖出状态

方法：

class Solution {
    public int maxProfit(int[] prices) {
        int len=prices.length;

        // 4种状态，0持股，1卖出，2保持卖出，3冷冻期
        int[][] dp=new int[len][4];

        // 初始化第一天持股
        dp[0][0]=-prices[0];
        dp[0][1]=0;
        dp[0][2]=0;
        dp[0][3]=0;

        for (int i=1;i<len;i++){
            dp[i][0]=Math.max(Math.max(dp[i-1][0],dp[i-1][2]-prices[i]),dp[i-1][3]-prices[i]);
            dp[i][1]=dp[i-1][0]+prices[i];
            dp[i][2]=Math.max(dp[i-1][2],dp[i-1][3]);
            dp[i][3]=dp[i-1][1];
        }

        return Math.max(Math.max(dp[len-1][1],dp[len-1][2]),dp[len-1][3]);
    }
}

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

题目链接：714. 买卖股票的最佳时机含手续费 - 力扣（LeetCode）

思路：和无限次买卖的股票问题二122. 买卖股票的最佳时机 II - 力扣（LeetCode）基本相同，只是在卖出股票的时候减去手续费

方法：

class Solution {
    public int maxProfit(int[] prices, int fee) {
        int len=prices.length;

        int[][] dp=new int[len][2];

        // 初始化第一天持股
        dp[0][0]=-prices[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]);
    }
}

4. 股票总结

dp数组设置不同的状态，根据前一天的不同状态推出当前状态然后取其中的最大值。