https://www.acwing.com/problem/content/description/137/
大意:给出长度为n的数组,求长度不超过m的连续子序列的和最大是多少。
思路:
- 固定右端点i,查找左端点j,如果k < j < i,并且sum[k] > sum[j]的话,那sum[i]-sum[j]一定更大,并且对于大于i的位置,都会去找j而不是k。
- 所有用单调队列维护一个前缀和的值随着下标增大而增大的序列,并且队列的大小不超过m,利用sum[i]-sum[q[l]]不断更新最值。
- 单调队列维护的下标都在[i-m,i-1]内。
code
#include <iostream>
#include <algorithm>
#include <stdio.h>
#include <queue>
//#include <bits/stdc++.h>
#define ll long long
const int N = 1e6+7;
const int mod = 1e9+7;
const ll ds = 1e15;
const double eps = 1e-8;
using namespace std;
int q[N];//q存下标
ll a[N];
void solve(){
int n,m,l = 1,r = 1;
cin >> n >> m;
for(int i = 1; i <= n; i++){
cin >> a[i];
a[i] = a[i-1]+a[i];
}
ll ans = -ds;
q[l] = 0;
for(int i = 1; i <= n; i++){
while(l <= r && q[l] < i-m) l++;//如果q[l]再区间[i-m,i-1]区间外,则++。
ans = max(ans,a[i]-a[q[l]]);
while(l <= r && a[q[r]] >= a[i]) r--;//大于a[i]并且下标大于q[l]的前缀和是没有意义的
q[++r] = i;
}
cout << ans << endl;
}
int main(){
// int t;
// cin >> t;
// while(t--)
solve();
//system("pause");
return 0;
}