题目描述
今天是小Z的生日,同学们为他带来了一块蛋糕。这块蛋糕是一个长方体,被用不同色彩分成了N个相同的小块,每小块都有对应的幸运值。
小Z作为寿星,自然希望吃到的第一块蛋糕的幸运值总和最大,但小Z最多又只能吃M小块(M≤N)的蛋糕。
吃东西自然就不想思考了,于是小Z把这个任务扔给了学OI的你,请你帮他从这N小块中找出连续的k块蛋糕(k≤M),使得其上的幸运值最大。
输入格式
输入文件cake.in的第一行是两个整数N,M。分别代表共有N小块蛋糕,小Z最多只能吃M小块。
第二行用空格隔开的N个整数,第i个整数Pi代表第i小块蛋糕的幸运值。
输出格式
输出文件cake.out只有一行,一个整数,为小Z能够得到的最大幸运值。
输入输出样例
输入 #1复制
5 2 1 2 3 4 5
输出 #1复制
9
输入 #2复制
6 3 1 -2 3 -4 5 -6
输出 #2复制
5
说明/提示
对20%的数据,N≤100。
对100%的数据,N≤500000,|Pi|≤500。 答案保证在2^31-1之内。
解释:使用单调队列(单调队列就是☞找一组无序的数在一定区间内最大值或最小值)
例如:1 10 -5 4 5 4.5 19 3,区间为 3,那么我们要找出 i-3 ----- i 区间的最值
那么单调递增队列就是如下
第一:存入1
2 : 10 > 1 存入10 队列: 1 10
3 : -5 < 10 10出-5进 1 比 -5 大 1出-5进 队列: -5
4: 队列 :-5 4
5: 队列:-5 4 5
6: 因为-5的位置使队列超出3的范围,移出 3,然后5 > 4.5,5出4.5进 队列:4 4.5
7:因为 4 的位置超出范围,将4移出,队列:4.5 19
8: 队列:3
本题就是使用数组来替代队列,在下面代码中 a数组表示的就是队列位置 而a[left]就是此区域的最小值
left ,right通过指向sum的位置来判断是否超出范围,而通过left++实现移出旧数,(将超出队列的范围的数删除)
right - - 控制排序(若新的数小于队列中的数,则将这个数通过right - - 放入合适位置)
# include <iostream>
# include <algorithm>
# include <cstring>
# include <math.h>
# include <stdio.h>
# include <vector>
# include <map>
using namespace std;
# define ll long long
ll sum[500100];
ll a[500100];
ll min1 = 1e9;
ll max1 = -min1;
int main(){
ll n,m;
cin >> n >> m;
ll num = 0;;
memset(sum,0,sizeof(sum));
for(ll i = 1; i <= n; i++){
cin >> num;
sum[i] = sum[i-1] + num;
}
ll right = 1;
ll left = 1;//队列
a[1] = 1;//下标指向1
for(ll i = 2; i <= n; ++i){
while(left <= right && a[left] < i - m)//当a指向的下标已经超过范围m;
{
left++;//将left左移,除去旧的小的数。
}
ll max2 = sum[i] - sum[a[left]];
max1 = max(max2,max1);
while(left <= right && sum[a[right]] >= sum[i]){
right--;//当数列的最新值大于了队列里的值,那么我们就要将这个最新值放到合适位置
//使其保持一个 递增 / 递减 队列。
}
right++;
a[right] = i;//将最新的满足要求的数的位置存放到队列中。
}
cout << max1 << endl;
return 0;
}