输入一个长度为n的整数序列,从中找出一段长度不超过m的连续子序列,使得子序列中所有数的和最大。
注意: 子序列的长度至少是1。
输入格式
第一行输入两个整数n,m。
第二行输入n个数,代表长度为n的整数序列。
同一行数之间用空格隔开。
输出格式
输出一个整数,代表该序列的最大子序和。
数据范围
1≤n,m≤3000001≤n,m≤300000
输入样例:
6 4
1 -3 5 1 -2 3
输出样例:
7
前缀和直接枚举o(n^2)会超时,所以用单调队列来做。对于三个下标k<j<i来说,若是sum(k)>=sum(j),即存在 sum(I)-sum(k)<=sum(i)-sum(j) 所以可以排除k,由此种思想可以得出做法:
1.每个下标都需要入队列
2.每个下标入队列时需要排除队列里面的“k”
(由于单调队列升序排序,所以从末尾开始向前删除就可以)
3.如果队列超出m大小,需要删除成员
这样每次只需要计算队列头的元素的前缀和,即复杂度到达了o(n)
(由于单调队列升序排序,所以队列头的前缀和之差最大,由于每次 i 的更替,需要每次都计算前缀和差)
实质:计算出每个 下标 的在[m,i]内的最大和
#include<iostream>
#include<algorithm>
using namespace std;
int a[300010],q[300010];
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
int n,m;
cin>>n>>m;
for(int i=1;i<=n;i++){
cin>>a[i];
a[i]+=a[i-1];
}
int ans=-0x3f3f3f;
int l=1,r=1;
for(int i=1;i<=n;i++){
while(l<=r&&q[l]<i-m) l++;
ans=max(ans,a[i]-a[q[l]]);
while(l<=r&&a[q[r]]>=a[i]) r--;
q[++r]=i;
}
cout<<ans;
return 0;
}