输入一个长度为n的整数序列,从中找出一段长度不超过m的连续子序列,使得子序列中所有数的和最大。
注意: 子序列的长度至少是1。
输入格式
第一行输入两个整数n,m。
第二行输入n个数,代表长度为n的整数序列。
同一行数之间用空格隔开。
输出格式
输出一个整数,代表该序列的最大子序和。
数据范围
1 ≤ n , m ≤ 300000 1≤n,m≤300000 1≤n,m≤300000
输入样例:
6 4
1 -3 5 1 -2 3
输出样例:
7
思路: 这道题目的解法比经典的连续子序列的和有点不一样,那个题目是用dp来解决,这道题目虽说也可以用dp来解决, 但是可能会复杂度较高,这里采用单调队列的方法来解决。 这里值得注意的是, 在第一个点进入队列时候,由于队列是空的, 所以要注意下特殊的hack数据。
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 300010, INF = 0x3f3f3f3f;
int q[N], s[N], n, m;
int main()
{
cin >> n >> m;
for (int i = 1; i <= n; i ++)
{
scanf("%d", s + i);
s[i] += s[i - 1];
}
int hh = 0, tt = -1, res = -INF;
for (int i = 1; i <= n; i ++)
{
if(hh <= tt && i - q[hh] > m) hh ++;
res = max(res, s[i] - s[q[hh]]); // 这里不设置队列为空的条件。
while (hh <= tt && s[i] <= s[q[tt]]) tt --;
q[++ tt] = i;
}
cout << res << endl;
return 0;
}