【123】. 买卖股票的最佳时机 III(最大允许2次交易)

【123】. 买卖股票的最佳时机 III(最大允许2次交易)


【123】. 买卖股票的最佳时机 III(最大允许2次交易)

题目描述

在这里插入图片描述

思路

用【通用方法秒杀股票问题】
dp[n][k][0~1]: 其中n为天数,k最到目前允许的最大交易次数,0~1表示持有或者不持有

找到状态转移方程:
dp[i][k][0] = max(dp[i-1][k][0], dp[i-1][k][1] + prices[i])
dp[i][k][1] = max(dp[i-1][k][1], dp[i-1][k-1][0] - prices[i])

定义base case:
dp[-1][k][0] = dp[i][0][0] = 0
dp[-1][k][1] = dp[i][0][1] = -infinity

    public int maxProfit(int[] prices) {
        if(prices == null || prices.length == 0){
            return 0;
        }
        int n = prices.length;
        int[][][] dp = new int[n][3][2];
        for(int i = 0; i < n ;i++){
            for(int k = 1; k <= 2; k++){
                //base case
                if(i-1 == -1){
                    dp[i][k][0] = 0;
                    dp[i][k][1] = -prices[i];
                    continue;
                }
                dp[i][k][0] = Math.max(dp[i-1][k][0],dp[i-1][k][1]+prices[i]);
                dp[i][k][1] = Math.max(dp[i-1][k][1],dp[i-1][k-1][0]-prices[i]);
            }
        }
        return dp[n-1][2][0];
    }
发布了55 篇原创文章 · 获赞 1 · 访问量 927
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 深蓝海洋 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览