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
100% 1≤N≤100000 1<=p<=q<=n
Key To Problem
dp+单调队列优化。
f[i]表示的状态为以第i天为假期的最后一天所奶牛们的最大享受指数。
先用sum数组求出前缀和,f[i]=sum[i]-min{sum[k]}(i-q<=k<=i-p)
单调队列维护单调递增区间,每次队首取出最小值。
code
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 100010
#define inf 1ll<<62
using namespace std;
typedef long long ll;
int n,p,q;
ll sum[N];
ll f[N];
int que[N];
int main()
{
scanf("%d%d%d",&n,&p,&q);
for(int i=1;i<=n;i++)
{
int x;
scanf("%d",&x);
sum[i]=sum[i-1]+x;
}
int head=1,tail=0;
ll ans=-inf;
for(int i=p;i<=n;i++)
{
while(head<=tail&&sum[que[tail]]>=sum[i-p])
tail--;
que[++tail]=i-p;
while(que[head]<i-q)
head++;
f[i]=sum[i]-sum[que[head]];
ans=max(ans,f[i]);
}
printf("%lld\n",ans);
return 0;
}