题意

给你一个长度为 n 的整数序列{A1,A2,⋯,An},要求从中找出一段连续的长度不超过 m 的非空子序列,使得这个序列的和最大。

输入格式

第一行为两个整数 n,m

第二行为 n 个用空格分开的整数序列,每个数的绝对值都小于 1000。

输出格式

仅一个整数,表示连续长度不超过 m 的最大非空子序列和。

思路

单项队列模板题。

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
#include <iostream>

using namespace std;

const int N = 2e5 + 10;

int n, m, res = -1<<30;
int a[N], s[N], q[N];

int main() {
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i ++ ) {
scanf("%d", &a[i]);
s[i] = s[i - 1] + a[i];
}

int head = 0, tail = 0;
for (int i = 1; i <= n; i ++ ) {
while (head <= tail && q[head] + m < i) head ++ ;
res = max(res, s[i] - s[q[head]]); // 更新最大前缀和
while (head <= tail && s[q[tail]] >= s[i]) tail --;
q[ ++ tail] = i; // 记录插入元素的下标
}

printf("%d\n", res);

return 0;
}