123. 买卖股票的最佳时机 III

在这里插入图片描述
和上一题的区别是这次最多只能交易两次。我的解法是给dp数组多加一列,用来标识当前状态交易次数。详情看注释,另外这里的话应该也可以进行空间优化,但是dp数组的话还是让人更好理解。

class Solution {
    public int maxProfit(int[] prices) {
        int len = prices.length;
        //第二列的0代表当前交易完不剩股票,1代表当前交易完剩有股票
        //第三列的n代表交易过的次数
        int[][][] dp = new int[len][2][3];
        //刚开始时第三列只能为0,第0天的股票可以选择买或者不买
        dp[0][0][0] = 0;
        dp[0][1][0] = -1 * prices[0];
        for (int i = 1; i < len; i++) {
        	//1. 交易0次但手上有股票:要么是在这次买的,要么就是之前就持有
            dp[i][1][0] = Math.max(dp[i - 1][0][0] - prices[i], dp[i - 1][1][0]);
			//2. 交易0次且不持有股票,只有一种情况,没买过股票
            dp[i][0][0] = dp[i - 1][0][0];
            //3. 交易过1次手上不持有股票,涉及到是这次交易的还是上次交易的,如果是上次交易的,那么当前i值至少为2,所以根据i的取值分情况讨论
            if (i >= 2)
                dp[i][0][1] = Math.max(dp[i - 1][1][0] + prices[i], dp[i - 1][0][1]);
            else
                dp[i][0][1] = dp[i - 1][1][0] + prices[i];
            //4. 交易过一次手上持有股票,同样的涉及是上次交易的还是这次交易的,另外还要设计手上持有的股票是这次就买了还是上次买的
            //dp[i][1][1] = dp[i - 1][1][0];其实这里有点违和,这个的意思是我在i-1的时候卖了股票,又买了i-1的股票,有种做无用功的感觉,但是这确实是可行的一种情况,我觉得还是要考虑
            if (i >= 2)
                dp[i][1][1] = Math.max(dp[i - 1][1][0], Math.max(dp[i - 1][1][1], dp[i - 1][0][1] - prices[i]));
            else
                dp[i][1][1] = dp[i - 1][1][0];
            //5. 交易过2次,手上也就无法持有股票了,涉及情况是第二次的交易是在这次还是在之前进行的,如果是在之前就交易完2次的话,i的值至少为3,因为至少要到i到2的时候才能交换两次
            if (i  >= 3)
                dp[i][0][2] = Math.max(dp[i - 1][0][2], dp[i - 1][1][1] + prices[i]);
            else if (i  >= 2)
                dp[i][0][2] = dp[i - 1][1][1] + prices[i];

        }
        //因为可能出现交换不满两次的情况,所以dp[len-1][0][1]也要考虑进来
        int res = Math.max(dp[len - 1][0][2], dp[len - 1][0][1]);
        return res >= 0 ? res : 0;
    }
}

在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值