leetcode 1425. 带限制的子序列和—(每日一难day19)

1425. 带限制的子序列和

给你一个整数数组 nums 和一个整数 k ，请你返回 非空 子序列元素和的最大值，子序列需要满足：子序列中每两个 相邻 的整数 nums[i] 和 nums[j] ，它们在原数组中的下标 i 和 j 满足 i < j 且 j - i <= k 。

数组的子序列定义为：将数组中的若干个数字删除（可以删除 0 个数字），剩下的数字按照原本的顺序排布。

示例 1：

输入：nums = [10,2,-10,5,20], k = 2
输出：37
解释：子序列为 [10, 2, 5, 20] 。

示例 2：

输入：nums = [-1,-2,-3], k = 1
输出：-1
解释：子序列必须是非空的，所以我们选择最大的数字。

示例 3：

输入：nums = [10,-2,-10,-5,20], k = 2
输出：23
解释：子序列为 [10, -2, -5, 20] 。

提示：

1 <= k <= nums.length <= 1e5
-1e4 <= nums[i] <= 1e4

解析

1. 如果没有限制我们应该怎么做？可以从前枚举每个数nums[i]，包含当前数的最大值我们用 f[i]表示 , 之后需要遍历 i 之前所有点的最大值，然后加上nums[i]。 f[i] = max( f[0…i] , 0 ) + nums[i] ;（max == 0时意味着只包含nums[i]） 这样我们就可以得到 到i结点为止的最大和。
2. 限制了结点范围我们应该怎么做呢 ？没限制时我们取的是f[i] = max( f[0…i] , 0 ) + nums[i]，如今限制了两个数的最大下标差，那么我们就不能从0开始枚举了。f[i] = max( f[i-k…i] , 0 ) + nums[i]；（从i-k开始，包含i-k）。
3. 还有一个问题，这样枚举由于k的范围很大，时间复杂度为O（N*K）会超时，因此还需要解决穷尽 i 的f[i]问题。
4. 可以发现一个现象，如果 i<j 时，f[i]<=f[j],那么我们是不是不用考虑 f[i] 了，f[j] 的下标靠近后边数，并且更大，所以优先级更高了，这样就引出了单调队列的问题。构造一个单调递减队列，如果 j 的值大于队列头部，那么就将头部 pop，最后将 j 加入。队列的尾部放了最大值，不满足就pop尾部元素，每次只需要取满足间隔的尾部就好。

code

class Solution {
public:
   int constrainedSubsetSum(vector<int>& nums, int k) {
       int n=nums.size();
       deque<int> dq;
       int f[n];
       f[0]=nums[0];
       dq.push_front(0);
       int res=nums[0];
       for(int i=1;i<n;i++){
       	// 找到符合间隔的最大尾部
           while(!dq.empty()&&dq.back()+k<i){
               dq.pop_back();
           }
           // 每次取符合条件的最大值
           f[i]=max(f[dq.back()],0)+nums[i];
           res=max(res,f[i]);
           // 加入当前元素
           while(!dq.empty()&&f[i]>=f[dq.front()]){
                   dq.pop_front();
           }
           dq.push_front(i);
       }
       return res;
   }
};