专题学习1 - J - 最大连续和

题意

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

输入格式

第一行为两个整数 n,m；

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

输出格式

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

思路

单项队列模板题。

代码

#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;
}