135. 最大子序和
输入一个长度为n的整数序列,从中找出一段长度不超过m的连续子序列,使得子序列中所有数的和最大。
注意: 子序列的长度至少是1。
输入格式
第一行输入两个整数n,m。
第二行输入n个数,代表长度为n的整数序列。
同一行数之间用空格隔开。
输出格式
输出一个整数,代表该序列的最大子序和。
数据范围1≤n,m≤300000
输入样例:
6 4
1 -3 5 1 -2 3
输出样例:
7
解题思路
前缀和+单调队列
其实就是用单调栈维护前缀和
- 计算区间和,就是说我们要查询[l,r]区间的总值,就会用到前缀和pre[r]−pre[l−1],pre[i]表示前 i 项的和
- 题目中的要求是求长度小于m的最大子序和,所以要满足:
1. r−l≤m,即区间的长度不可以超过m
2. pre[r]−pre[l]尽可能大 - 这道题用普通dp明显不行,时间复杂度最坏O(n2),所以就要用的单调队列进行优化,单调队列需要维护单调性要使用双端队列deque,因为其中需要头部弹出以及尾部弹出,用来保存数组下标
- 要想满足是最优解即满足:下标递增,pre[i]递增
- 即对于两个下标k,j,如果k<j<i,pre[k]≧pre[j],那么对于所有大于等于i的右端点,k一定不是最优选择,这是因为不但pre[k]不小于pre[j](pre[i]-pre[j]≧pre[i]-pre[k]),而且j离i更近,长度更不容易超过m
- 具体步骤
- 如果i - 队首下标>m,则弹出队首元素,经过一系列此操作后队首元素就是对于i的最优左端点
- ans不断更新子序列和的最大值
- 不断删除队尾的元素直至队尾对应的前缀和小于 i 的前缀和(如果队尾的元素不可能有将要入队的元素优,也需抛出)
- 再把 i 做为一个新的左端点决策入队,队列中是可供选择的子序列左端点
具体操作见代码
附上代码
#include<bits/stdc++.h>
using namespace std;
const int INF=0x3f3f3f3f;
const int N=3e5+5;
int a[N],pre[N];
deque<int> q;
int main(){
int n,m;
cin>>n>>m;
for(int i=1;i<=n;i++){
cin>>a[i];
pre[i]=pre[i-1]+a[i];
}
int ans=-INF;
q.push_back(0);
for(int i=1;i<=n;i++){
while(q.size()&&q.front()<i-m)
q.pop_front();//如果队头已经无法满足区间长度小于m的话,过期,抛出
ans=max(ans,pre[i]-pre[q.front()]);//取当前最优值
while(q.size()&&pre[q.back()]>=pre[i])
q.pop_back();//如果说队尾大于当前值的话,非最优解,抛出
q.push_back(i);//入队
}
cout<<ans<<endl;
return 0;
}