题意
给你一个长度为 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; }
|