题目描述
今天是小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之内。
思路
n个数中取连续的k(k<=m)个数,输出其中所有可能的最大值。
#include <stdio.h>
#include <iostream>
#include <deque>
#define linf -2e9-7
using namespace std;
int n,m,s,sum[500001];
struct node
{
int v,position;
}a[500001];
deque<node> dq;
signed main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
register int i,j;
cin>>n>>m;
for(i=1;i<=n;i++)
{
cin>>a[i].v;
sum[i]=sum[i-1]+a[i].v;//前缀和
a[i].position=i;
}
for(i=1;i<=n;i++)
{
while(!dq.empty() && dq.front().position<i-m)//不在区间的要出队
dq.pop_front();
s=max(s,sum[i]-sum[dq.front().position]);//取最大值
while(!dq.empty() && sum[i]<=sum[dq.back().position])//维护单调递增的序列
dq.pop_back();
dq.push_back(a[i]);//元素入队
}
cout<<s<<endl;
return 0;
}
来源:https://blog.csdn.net/Apro1066/article/details/99474300