1. 程式人生 > 其它 >專題學習1 - J - 最大連續和

專題學習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;
}