最大子序和
链接:https://ac.nowcoder.com/acm/contest/1006/D
题目描述
输入一个长度为n的整数序列,从中找出一段不超过m的连续子序列,使得整个序列的和最大。
例如 1,-3,5,1,-2,3
当m=4时,S=5+1-2+3=7
当m=2或m=3时,S=5+1=6
输入描述:
第一行两个数n,m(1≤n,m≤300000)
第二行有n个数,要求在n个数找到最大子序和
输出描述:
一个数,数出他们的最大子序和
示例
输入
6 4
1 -3 5 1 -2 3
输出
7
题解
- 前缀和
- 单调队列
因为我们要查询 [l,r] 区间的值,我们显然会想到前缀和
a[r] − a[l] 表示 a[l, r] 的和
那么我们就是要求两个位置l, r
对于这两个位置要求- r − l ≤ M 即区间的大小不可以超过M
- a[r] − a[l] 尽量大
我们可以 确定 r 为 i 从 1 ~ n 遍历一遍
然后为了使 a[r] − a[l] 尽量大,也就是必须使得当 r == i 时,a[i] > a[l]
构成单调的队列,每次遍历 i 取最大值即可。
#include<stdio.h>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N = 3e5 + 10;
int n, m;
int q[N];
ll a[N];
int main(){
scanf("%d%d",&n,&m);
for(int i=1; i<=n; ++i){
scanf("%lld",&a[i]);
a[i] += a[i-1];
}
int head = 0, tail = 0;
ll ans = -999999;
for(int i=1; i<=n; ++i){
if(i - q[head] > m) head++;
ans = max(ans, a[i] - a[q[head]]);
while(head <= tail && a[q[tail]] >= a[i]) tail--;
q[++tail] = i;
}
printf("%lld\n",ans);
return 0;
}