leetcode123.买卖股票的最佳时机III：一维动态规划

最新推荐文章于 2024-06-26 21:03:37 发布

wmtmw

最新推荐文章于 2024-06-26 21:03:37 发布

阅读量142

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/yinianxx/article/details/108464735

版权

leetcode 专栏收录该内容

87 篇文章 0 订阅

订阅专栏

题目
在这里插入图片描述
思路一：三维动态规划
dp[i][j][k]：
i表示第几天，
j表示是否持有股票，0表示不持有，1表示持有
k表示第几次交易，取值0，1，2

具体代码

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

思路二：二维动态规划
dp[i][j]:
j取值为0~4，0表示未交易，1表示买入一次，2表示卖出一次，3表示买入两次，4表示卖出2次

具体代码

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

思路三：一维动态规划
其实就是从二维动态规划演进过来，初始化的值发生了改变。

具体代码

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

参考链接