输入一个长度为 n 的整数序列,从中找出一段长度不超过 m 的连续子序列,使得子序列中所有数的和最大。
注意: 子序列的长度至少是 1。
输入格式
第一行输入两个整数 n,m。
第二行输入 n 个数,代表长度为 n 的整数序列。
同一行数之间用空格隔开。
输出格式
输出一个整数,代表该序列的最大子序和。
数据范围
1≤n,m≤300000
输入样例:
6 4 1 -3 5 1 -2 3
输出样例:
7
分析:
求一段连续的长度不超过m的序列最大和
单调队列套前缀和
代码:
#include <bits/stdc++.h>
using namespace std;
const int N = 300010;
int n,m;
long long sum[N];
int q[N], hh, tt;
int main()
{
cin >> n >> m;
long long ans = -1e18;
for (int i = 1; i <= n; i ++ )
{
int x;
scanf("%d", &x);
sum[i] = sum[i - 1] + x;
// 在一个到当前位置i来说,长度不超过m+1的非严格单调递增的序列中,减去第一个位置的sum
// while (hh <= tt && i - q[hh] > m) hh ++;
// 换成更好理解的方式
while (hh <= tt && q[hh] < i - m) hh ++;
// for (int j = hh; j <= tt; j ++ ) printf("%d ", q[i]); puts("");
ans = max(ans, sum[i] - sum[q[hh]]);
// 维护单调序列
while (hh <= tt && sum[q[tt]] > sum[i]) tt --;
q[ ++ tt] = i;
}
cout << ans << endl;
}