题目
题目描述
输入一个长度为 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。
第二行有 n个数,要求在 n 个数找到最大子序和。
输出
一个数,数出他们的最大子序和
相关知识点
- 使用c++ STL模板库中的双端队列deque实现这道题中需要用的单调队列
- deque中提供了几个常用的接口如下:
front():返回容器的第一个元素
back():返回容器的最后一个元素
empty():判断容器是否为空
size():返回容器中实际拥有的元素个数
push_back():在末尾位置插入元素
pop_back():在末尾位置删除元素
push_front():在开头位置插入元素
pop_front():在开头位置删除元素
题目分析
- 求长度不超过m的最大子序和可以通过维护一个长度为m的单调递增队列来求解,用单调递增队列的最后一个元素减第一个元素得到的就是这个最大子序和
代码及注释
#include <iostream>
#include <string>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <map>
#include <vector>
#include <deque>
using namespace std;
#define MAX_N 300000
int arr[MAX_N + 5] = {0};
int sum[MAX_N + 5] = {0};
int main() {
int n, m;
cin >> n >> m;
for (int i = 1; i <= n; i++) { //从1号开始,0号位置零,因为后面元素是通过与0号位置相减来求子序和的
cin >> arr[i];
sum[i] = sum[i - 1] + arr[i]; //构造和的数组
}
deque<int> q; //定义一个双端队列,队列里维护的元素是原始数组元素的下标
q.push_back(0);
int ans = sum[1];
for (int i = 1; i <= n; i++) {
ans = max(ans, sum[i] - sum[q.front()]); //这一句必须放到前面
while (q.back() > 0 && sum[q.back()] >= sum[i]) { //下标为了0的元素是为了相减方便加上的,本身不在真正的数据中,因此这里不需要弹出
q.pop_back();
}
if (i - m == q.front()) {// 这里注意当要添加的元素与队列最大长度的差与队首元素相等,就应该弹出队,而不是根据当前队列里实际拥有元素个数
q.pop_front();
}
q.push_back(i); //每向后走一位,都要把最新的下标添加到队列的末尾
}
cout << ans << endl;
return 0;
}