每日随机一题 leetcode1425. 带限制的子序列和

题:
在这里插入图片描述


思:

dp[i]表示右端点为i的最大的子序列和
dp[0] = nums[0]
dp[i] = max(max(dp[i-Q])+nums[i],nums[i]) ; Q = 1,2,…,k
化简得 dp[i] = max(dp[i-Q],0) + nums[i]; Q = 1,2,…,k
返回dp里最大的值

上面的方法会超时,看了答案的思路,需要使用单调队列(单调栈)进行维护,避免多余的运算
单调栈原理:
如果dp[i] > dp[i-1] 则每次挑选max的时候,就不要去和dp[i-1]比了,因此用栈来存放可以用来比的数


码:

public int constrainedSubsetSum(int[] nums, int k) {
        int length = nums.length;
        int[] dp = new int[length];
        dp[0] = nums[0];
        int ans = dp[0];
        // 使用单调队列(因为要从队首获取当前最大的子序列和,队尾插入新元素)
        Deque<Integer> deque = new ArrayDeque<Integer>();
        deque.addLast(0);
        for (int i = 1; i < dp.length; i++) {
            int max = 0;
            // 获取队首元素,队首元素是当前数前面为止的最大子序列和,但是如果他们的距离超过k,就先更新队列,再获取队首元素
            while(!deque.isEmpty() && i-deque.peekFirst() > k){
                deque.removeFirst();
            }
            // 获取队首元素
            int firstValue = dp[deque.peekFirst()];
            max = Math.max(0,firstValue);
            dp[i] = max+nums[i];
            ans = Math.max(ans,dp[i]);
            // 维护队列的单调性,如果新加入的这个dp值,比原来队首的要大,则清空队列,让队列里只剩下一个元素,就是dp[i]
            // 不能这么做,这么做无法保证队首是最大的子序列和所对应的下标
            // 正确做法:
            // 每次新加入一个元素以后,要保证当前队列里,没有比这个元素要大的子序列和
            while (!deque.isEmpty() && dp[i] > dp[deque.peekLast()]){
                deque.removeLast();
            }
            //最后,加入这个元素
            deque.addLast(i);
            /*
            if(dp[i] > firstValue){
                deque.clear();
                deque.addLast(i);
            }else {
                // 否则,将这个dp值加入
                deque.addLast(i);
            }
             */

        }
        return ans;
    }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值