输入一个长度为 n 的整数序列,从中找出一段长度不超过 m 的连续子序列,使得子序列中所有数的和最大。
注意: 子序列的长度至少是 1。
思路:先求前缀和。从1遍历到n将其作为队列尾算出其前m个前缀和差值的最大值。及求其前m个前缀和中最小值
所用技巧
单调队列 随着遍历队列前的数出队列队列后的数入队列。可以发现当队列中一个数的前一个数大于或等于这个数则前一个数无用(这个数先出而且去最小前缀和时因为其前一个数的存在不会被选取)。如此队列形成了一个单调增队列。队列中最小前缀和是对列头。
因为每个数只会进出队列一次且取最小前缀和操作时间复杂度为o(1)所以其总时间复杂度并不大。
代码如下
#include <iostream>
#include <algorithm>
#include <limits.h>
using namespace std;
const int N = 3e5 + 10;
int s[N], q[N];
int main(){
int n, m;
cin >> n >> m;
for(int i = 1;i <= n;i ++){
scanf("%d",&s[i]);
s[i] += s[i - 1];
}
int res = INT_MIN, hh = 0, tt = 0;// 这个hh和tt的选取是将s[0]作为队列头
for(int i = 1;i <= n;i ++){
while(i - q[hh] > m) hh ++;//这里while中减相当于是前缀和减所以是大于号
res = max(res, s[i] - s[q[hh]]);
while(hh <= tt && s[q[tt]] >= s[i]) tt --;
q[++ tt] = i;
}
cout << res << endl;
}