bzoj 4385: [POI2015]Wilcze doły

bzoj 4385: [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。

 

Source

鸣谢Claris

---

solution

我好弱，想了好久都没想出来

考虑枚举右端点，那么左端点一定是单调的。

现在问题是怎么确定最小的左端

于是我们希望让删去的那一段尽量大

删去的那一段应该也是单调递减的（如果位置较后的还更大，那位置前的一定不优）

单调队列维护即可

#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#define maxn 2000006
#define ll long long
using namespace std;
ll n,p,d,s[maxn],a[maxn]; 
ll q[maxn],f,r,ans,j;
ll Max(){
	while(f<r&&q[f+1]-d<=j)f++;
	if(f==r)return 1e16;
	else return s[q[f+1]]-s[q[f+1]-d];
}
int main(){
	cin>>n>>p>>d;
	for(int i=1;i<=n;i++){scanf("%lld",&a[i]);s[i]=s[i-1]+a[i];}
    j=0;f=0;
	for(int i=d;i<=n;i++){
    	while(f<r&&s[q[r]]-s[q[r]-d]<=s[i]-s[i-d])r--;
    	q[++r]=i;
		while(s[i]-s[j]-Max()>p)j++;
		ans=max(ans,i-j);
	}
	cout<<ans<<endl;
	return 0;
}