题目
输入一个长度为n的整数序列,从中找出一段长度不超过m的连续子序列,使得子序列中所有数的和最大。
输入格式
第一行输入两个整数n,m。
第二行输入n个数,代表长度为n的整数序列。
同一行数之间用空格隔开。
输出格式
输出一个整数,代表该序列的最大子序和。
数据范围
1≤n,m≤300000
输入样例:
6 4
1 -3 5 1 -2 3
输出样例:
7
区间和 -> 前缀和相减sum[r]-sum[l-1]
找到i,j两个位置,ans = max(sum[i]-sum[j]) if i-j<=m
枚举i ,找到 min(sum[j]) and x∈[i-m,i-1]
最优决策集合:下标递增,sum[j]递增 ->用双端队列保存下标
维护:
1.队头出队 if 队头与i 距离 > m
2.此时队头即为最好的 j
3.不断pop队尾,直至sum[队尾] < s[i] ,i入队
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=3e5+5;
deque<int>q;
int n,m,ans;
int a[N],sum[N];
int main(){
ios::sync_with_stdio(false);
cin.tie(0),cout.tie(0);
q.push_back(0);
cin>>n>>m;
for(int i=1;i<=n;++i)cin>>a[i],sum[i]=sum[i-1]+a[i];
for(int i=1;i<=n;++i){
while(!q.empty()&&q.front()<i-m)q.pop_front();
ans=max(ans,sum[i]-sum[q.front()]);
while(!q.empty()&&sum[q.back()]>=sum[i])q.pop_back();
q.push_back(i);
}
cout<<ans;
return 0;
}