4385: [POI2015]Wilcze doły (单调队列)

http://www.lydsy.com/JudgeOnline/problem.php?id=4385

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


维护个前缀和f[i]和s[i],s[i]表示从i开始,长度为d的区间的和。
然后单调队列维护一个区间,使这个区间满足:f[j] - f[i] - MAX[s[x]]

单调队列维护的是 s[x]的最大值。
然后再根据区间的情况使左端点移动…

%lld 会 AC 。
%I64d 会WA 。


#include 
#include 
#include 
#include 
using namespace std;

typedef long long LL;
const int N = 2*1e6 + 100;
LL f[N],que[N],s[N];
int pos[N];
int main()
{
        int n,d;
        LL x,p;
        scanf("%d%lld%d",&n,&p,&d);
        for(int i=1;i<=n;i++) scanf("%lld",&x), f[i] = f[i-1] + x;
        for(int i=1;i<=n-d+1;i++) s[i] = f[i+d-1] - f[i-1];        

        int head = 1, tail = 0, l = 0;
        int ans = d;
        for(int i=d , j = 1; i<=n; i++, j++)
        {
                while( head <=tail && que[tail]<=s[j] ) tail--;
                que[++tail] = s[j];
                pos[tail] = j;

                while( l p )
                {
                        l++;
                        if( l >= pos[head] ) head++;
                }
                ans = max( ans, i - l );
        }
        printf("%d\n",ans);
        return 0;
}

你可能感兴趣的:(*Dynamic,Programming)