题目描述
今天是小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的数列中,找出一个长度≤M的子段,使得它的和是最大的。
第i个元素结尾的子段,最大的子段和P(i)可以表示为
P[i]=max{sum[i]-sum[j],j属于[i-M,i-1]}
这样去做的是O(nm)是肯定会T 的
实际上这题和Best Cow Fences(POJ2018)中单调队列的优化是类似的
转化一下:
P[i]=sum[i]-min{sum[j]},j属于[i-M,i-1] ,那么我们预处理出这个sum[i-M,i-1]的最小值就可以了,而且区间是固定长的,就是类似一个滑动窗口,这里可以用单调队列去优化
那这里我用了ST表预处理,然后在查的时候知道了范围,就可以直接找范围内的最小值
#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=500010;
typedef long long LL;
LL a[maxn],sum[maxn];
LL f[maxn][20];
LL n,m;
inline LL read()
{
LL x=0,f=1;char ch=getchar();
while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}
while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
return x*f;
}
void init()
{
for(LL i=1;i<=n;i++) f[i][0]=sum[i];
LL t=log(n)/log(2)+1;
for(LL j=1;j<t;j++)
{
for(LL i=1;i<=n-(1<<j)+1;i++)
{
f[i][j]=min(f[i][j-1],f[i+(1<<(j-1))][j-1]);
}
}
}
LL ask(LL l,LL r)
{
LL k=log(r-l+1)/log(2);
return min(f[l][k],f[r-(1<<k)+1][k]);
}
int main(void)
{
n=read();m=read();
for(LL i=1;i<=n;i++) a[i]=read();
for(LL i=1;i<=n;i++) sum[i]=sum[i-1]+a[i];
init();//ST表初始化
LL ans=-0x7f7f7f7f7f7f7f;
for(LL i=1;i<=n;i++)
{
//LL l=i-m;
//if(l<0) l=0;
ans=max(ans,sum[i]-ask(max((LL)0,i-m),i));
// cout<<"i=:"<<i<<'\t'<<ask(1,n)<<endl;
}
cout<<ans<<endl;
return 0;
}