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
解析
- 如果没有限制我们应该怎么做?可以从前枚举每个数nums[i],包含当前数的最大值我们用 f[i]表示 , 之后需要遍历 i 之前所有点的最大值,然后加上nums[i]。 f[i] = max( f[0…i] , 0 ) + nums[i] ;(max == 0时意味着只包含nums[i]) 这样我们就可以得到 到i结点为止的最大和。
- 限制了结点范围我们应该怎么做呢 ?没限制时我们取的是f[i] = max( f[0…i] , 0 ) + nums[i],如今限制了两个数的最大下标差,那么我们就不能从0开始枚举了。f[i] = max( f[i-k…i] , 0 ) + nums[i];(从i-k开始,包含i-k)。
- 还有一个问题,这样枚举由于k的范围很大,时间复杂度为O(N*K)会超时,因此还需要解决穷尽 i 的f[i]问题。
- 可以发现一个现象,如果 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;
}
};