Post

AcWing135 - 最大子序和

输入一个长度为 n 的整数序列,从中找出一段长度不超过 m 的连续子序列,使得子序列中所有数的和最大。

注意: 子序列的长度至少是 1。

输入格式

第一行输入两个整数 n,m。

第二行输入 n 个数,代表长度为 n 的整数序列。

同一行数之间用空格隔开。

输出格式

输出一个整数,代表该序列的最大子序和。

数据范围

1≤n,m≤300000, 保证所有输入和最终结果都在 int 范围内。

输入样例:

1
2
6 4
1 -3 5 1 -2 3

输出样例:

1
7

单调队列优化

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
#include <iostream>
using namespace std;
const int N = 300010;

int n, m;
int s[N], q[N];

int main() {
  scanf("%d%d", &n, &m);
  for (int i = 1; i <= n; i++) {
    scanf("%d", &s[i]);
    s[i] += s[i - 1];
  }
  
  int res = s[1];
  int hh = 0, tt = 0;
  for (int i = 1; i <= n; i++) {
    if (q[hh] < i - m) hh++;
    res = max(res, s[i] - s[q[hh]]);
    while (hh <= tt && s[q[tt]] >= s[i]) tt--;
    q[++tt] = i;
  }
  
  printf("%d\n", res);
  return 0;
}
This post is licensed under CC BY 4.0 by the author.