專題學習1 - J - 最大連續和
阿新 • • 發佈:2022-03-28
題意
給你一個長度為 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; }