【BZOJ4385】[POI2015]Wilcze doły 单调栈+双指针法

【BZOJ4385】[POI2015]Wilcze doły

Description

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

Input

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

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。

题解:易知我们向[l,r]中加入一个数r+1,得到的[l,r+1]的区间和(修改后)肯定比[l,r]要大,这显然满足双指针法的条件

所以我们还要动态维护[l,r]中,和最大的长度为d的区间,也就是维护区间最大值,这显然又可以用单调队列来搞定,所以这道题将两个单调数据结构套在了一起,不过并不难

#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;
const int maxn=2000010;
int n,d,ans;
int q[maxn];
long long p,s[maxn],v[maxn];
int main()
{
    scanf("%d%lld%d",&n,&p,&d);
    int i,h=1,t=0,j=0;
    for(i=1;i<=n;i++)    scanf("%d",&s[i]),s[i]=s[i-1]+s[i];
    ans=d;
    for(i=d;i<=n;i++)
    {
        while(h<=t&&s[i]-s[i-d]>=s[q[t]]-s[q[t]-d])   t--;
        q[++t]=i;
        while(h<=t&&s[i]-s[j]-s[q[h]]+s[q[h]-d]>p)
        {
            j++;
            if(q[h]<j+d) h++;
        }
        ans=max(ans,i-j);
    }
    printf("%d",ans);
    return 0;
}
posted @ 2017-05-15 10:22  CQzhangyu  阅读(390)  评论(0编辑  收藏  举报