[BZOJ4385] [POI2015]Wilcze doły

题目描述

Description

给定一个长度为n的序列,你有一次机会选中一段连续的长度不超过d的区间,将里面所有数字全部修改为0。
请找到最长的一段连续区间,使得该区间内所有数字之和不超过p。

Input

第一行包含三个整数n,p,d(1<=d<=n<=2000000,0<=p<=10^16)。
第二行包含n个正整数,依次表示序列中每个数wi

Output

包含一行一个正整数,即修改后能找到的最长的符合条件的区间的长度。

Sample Input

9 7 2
3 4 1 9 4 1 7 1 3

Sample Output

5

HINT

将第4个和第5个数修改为0,然后可以选出区间[2,6],总和为4+1+0+0+1=6。

题目分析

先要预处理出每个点向左拓展出的最长(区间和不超过d)的区间和
我们想:假设我们选定了一段区间 如果这段区间和减去这段区间内的 最大的预处理 小于等于p 我们可以继续选对吧
然后我们用双指针去维护这个过程 同时单调队列维护一下这个区间内的最大预处理即可

#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
int n,d,ans;
int q[2000010];
long long p,sum[2000010];
int main()
{
    scanf("%d%lld%d",&n,&p,&d);
    int h=1,t=0,j=0;
    for(int i=1;i<=n;i++)    scanf("%d",&sum[i]),sum[i]=sum[i-1]+sum[i];
    ans=d;
    for(int i=d;i<=n;i++)
    {
        while(h<=t&&sum[i]-sum[i-d]>=sum[q[t]]-sum[q[t]-d])   t--;
        q[++t]=i;
        while(h<=t&&sum[i]-sum[j]-sum[q[h]]+sum[q[h]-d]>p)
            if(q[h]<(++j)+d) h++;
        ans=max(ans,i-j);
    }
    printf("%d",ans);
    return 0;
}

发表评论

邮箱地址不会被公开。 必填项已用*标注