首先,把数都弄成前缀和sum[i],节约时间;
枚举右端点i,当i固定的时候,问题就变成了,寻找一个左端点j,j在区间
[i-m,i-1]内,并且sum[j]最小;
使用一个单调队列来完成此题,我们从左往右扫一遍右端点,对每个i:
1.判断当前队头的数与i的距离是否大于m,大于就把队头干掉
2.此时队头就是右端点为i时的最佳答案(因为,不是最佳的答案在每次第三步更新sum[i]进队的时候都被干掉了)
3.不断删除队尾直到队尾的s值小于s[i](因为如果一个s值比s[i]大,他的位置又比i靠左,那么他是没有存在意义的),将s[i]入队;
两种写法:
1.数组模拟
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
long long sum[300005];
int main()
{
int n,m,i,j,l,r,t;
cin>>n>>m;
for(i=1;i<=n;i++){
scanf("%d",&t);
sum[i]=sum[i-1]+t;
}
long long ans=1<<31;//q保存的是数列中数的位置
l=r=1;int q[300005];q[1]=0;//第一个入列的一定是0,因为前缀和存在s[i]-s[0]的情况,代表的是第1到个数
for(i=1;i<=n;i++){
while(l<=r&&q[l]<i-m)l++;//当队首元素的距离超过m到时候pop
ans=max(ans,sum[i]-sum[q[l]]);//更新答案
while(l<=r&&sum[q[r]]>=sum[i])r--;//当队尾的数值不如当前数值时,pop队尾
q[++r]=i;
}
cout<<ans<<endl;
return 0;
}
2.使用stl双端队列
#include<cstdio>
#include<iostream>
#include<deque>
using namespace std;
deque<pair<int,int> >que;
int sum[300005];
int main()
{
int n,m,i,j,k;
cin>>n>>m;
for(i=1;i<=n;i++){
int t;
scanf("%d",&t);
sum[i]=sum[i-1]+t;
}
que.push_front(pair<int,int>(sum[0],0));
int ans=1<<31;
for(i=1;i<=n;i++){
while(que.size()&&que.front().second<i-m)
que.pop_front();
if(que.size())ans=max(ans,sum[i]-que.front().first);
while(que.size()&&que.back().first>sum[i])
que.pop_back();
que.push_back(pair<int,int>(sum[i],i));
}
cout<<ans<<endl;
return 0;
}