算法练习Day50|● 123.买卖股票的最佳时机III ● 188.买卖股票的最佳时机IV

 1class Solution {
 2    public int maxProfit(int[] prices) {
 3
 4        // 根据买入卖出定义多种状态：0表示不操作，1表示第一次持有，2表示第一次不持有，3表示第二次持有，4表示第2次不持有
 5        int[][] dp = new int[prices.length][5];
 6        dp[0][0] = 0;
 7        dp[0][1] = -prices[0];
 8        dp[0][2] = 0;
 9        dp[0][3] = -prices[0];
10        dp[0][4] = 0;
11
12        for (int i = 1; i < prices.length; i++) {
13            // 第一次持有，前一天就持有 或 第i天买入持有
14            dp[i][1] = Math.max(dp[i - 1][1], -prices[i]);
15            // 第一次不持有，前一天不持有 或 第i天不持有
16            dp[i][2] = Math.max(dp[i - 1][2], dp[i - 1][1] + prices[i]);
17            // 第二次持有，前一天第二次持有 或 前一天不持有第i天持有
18            dp[i][3] = Math.max(dp[i - 1][3], dp[i - 1][2] - prices[i]);
19            // 第二次不持有，前一天第二次不持有 或前一天第二次持有第i天卖出
20            dp[i][4] = Math.max(dp[i - 1][4], dp[i - 1][3] + prices[i]);
21        }
22        return Math.max(dp[prices.length - 1][2], dp[prices.length - 1][4]);
23    }
24}
25

 1class Solution {
 2    public int maxProfit(int k, int[] prices) {
 3
 4        // 定义dp[][][] 数组，[天数][交易次数][是否持有股票]
 5        int len = prices.length;
 6        int[][][] dp = new int[len][k + 1][2];
 7
 8        // 初始化dp数组
 9        // 初始化所有的交易次数是为确保 最后结果是最多 k 次买卖的最大利润
10        for (int i = 0; i <= k; i++) {
11            dp[0][i][1] = -prices[0];
12        }
13
14        for (int i = 1; i < len; i++) {
15            for (int j = 1; j <= k; j++) {
16                // 0 表示不持有股票
17                // 1 表示持有股票
18                dp[i][j][0] = Math.max(dp[i - 1][j][0], dp[i - 1][j][1] + prices[i]);
19                dp[i][j][1] = Math.max(dp[i - 1][j][1], dp[i - 1][j - 1][0] - prices[i]);
20            }
21        }
22        return dp[len - 1][k][0];
23    }
24}
25