Description
经过几个月辛勤的工作,FJ决定让奶牛放假。假期可以在1…N天内任意选择一段(需要连续),每一天都有一个享受指数W。但是奶牛的要求非常苛刻,假期不能短于P天,否则奶牛不能得到足够的休息;假期也不能超过Q天,否则奶牛会玩的腻烦。FJ想知道奶牛们能获得的最大享受指数。
Input
第一行:N,P,Q.
第二行:N个数字,中间用一个空格隔开,每个数都在longint范围内。
Output
一个整数,奶牛们能获得的最大享受指数。
Sample Input
5 2 4
-9 -4 -3 8 -6
Sample Output
5
Hint
50% 1≤N≤10000
100% 1≤N≤100000
1<=p<=q<=n
样例解释:选择第3-4天,享受指数为-3+8=5。
分析
这题也是比较纯粹的单调队列。具体做法如下:
看到区间的问题首先肯定是想到求前缀和,我们把1~k的和记为sum[k],可以得到sum[i] = sum[i - 1] + a[i],[l,r]的和即为sum[r] - sum[l - 1](这里视sum[0] = 0)。我们假设选择的区间为[l,r]且r固定,可知r-q+1≤l≤r-p+1,若要使[l,r]区间的值最大,则sum[l - 1]需最小,才可使得sum[r] - sum[l - 1]最小,当i右移一位到i+1,因为p,q为给定不变的值,对应寻找最小sum[l-1]的区间也右移一位。
在单调队列中,有两种情况可以出队,一种情况可以入队。
先说出队的:
- 目前队尾的那个值没有新来的那个值那么优(有时候是更大有时候是更小,看情况做出判断)
- 队头的那个数已经超过的区间的最大判断范围,则从队头直接出队(h++)
入队的:
- 准备要入队的那个值做完比较成功后,从队尾入队。
注意:代码中第16行的初始化!
1.之所以要
h
=
1
,
t
=
0
h=1,t=0
h=1,t=0是因为队头指针一开始指向1,而当时队列中却没有数,也就是没有队尾,所以t=0。直到几个数都通过第一层for循环进来之后才开始有判断出队的操作。
2.ans变量的初始化为负数(-2147483647)是因为在题目中享受指数比较出来不一定是正数,也有可能是负数,所以就用int范围内最小的负数作为初始值。
上代码
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
long long n,p,q,a[100001],que[1000001],sum[100001];
int main()
{
cin>>n>>p>>q;
a[0]=0;
for(int i=1;i<=n;i++)
{
cin>>a[i];
sum[i]=sum[i-1]+a[i];
}
long long h=1,t=0,ans=-2147483647; //!!!!!!
for(int i=p;i<=n;i++)
{
while(h<=t&&sum[que[t]]>=sum[i-p])//如果i-p的比队尾的更优,就去尾
{
t--;
}
t++;
que[t]=i-p;//把这个更优的从队尾加入队列
while(h<=t&&que[h]<i-q)//如果队头元素超出统计范围
{
h++;
}
ans=max(ans,sum[i]-sum[que[h]]);//统计&更新
}
cout<<ans;
return 0;
}