代码随想录二刷day50

提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档

文章目录

  • 前言
  • 一、力扣123. 买卖股票的最佳时机 III
  • 二、力扣188. 买卖股票的最佳时机 IV


前言


一、力扣123. 买卖股票的最佳时机 III

class Solution {
    public int maxProfit(int[] prices) {
        int n = prices.length;
        int[][] dp = new int[n][4];
        dp[0][0] = -prices[0];
        dp[0][2] = -prices[0];
        for(int i = 1; i < n; i ++){
            dp[i][0] = Math.max(dp[i-1][0], -prices[i]);//第一次持有
            dp[i][1] = Math.max(dp[i-1][1], dp[i-1][0] + prices[i]);//第一次卖出
            dp[i][2] = Math.max(dp[i-1][2], dp[i-1][1] - prices[i]);//第二次买入
            dp[i][3] = Math.max(dp[i-1][3], dp[i-1][2] + prices[i]);//第二次卖出
        }
        return Math.max(dp[n-1][1], dp[n-1][3]);
    }
}

二、力扣188. 买卖股票的最佳时机 IV

class Solution {
    public int maxProfit(int k, int[] prices) {
        int n = prices.length;
        int[][][] dp = new int[n][k+1][2];//[]天数  []第几次买卖 []持有还是不持有//0持有1不持有
        for(int i = 0; i <= k;i ++){
            dp[0][i][0] = -prices[0];
        }
        for(int i = 1; i < n; i ++){
            for(int j = 1; j <= k; j ++){
                dp[i][j][0] = Math.max(dp[i-1][j][0], dp[i-1][j-1][1] - prices[i]);
                dp[i][j][1] = Math.max(dp[i-1][j][1], dp[i-1][j][0] + prices[i]);
            }
        }
        return dp[n-1][k][1];
    }
}

你可能感兴趣的:(算法,数据结构,leetcode,java,动态规划)