首页 > 代码库 > bzoj4385 & POJ2015 Wilcze do?y
bzoj4385 & POJ2015 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
3 4 1 9 4 1 7 1 3
Sample Output
5
Solution
i-j<=d没有任何意义
左端点要尽量靠左,随着右端点向右移动,左端点一定也向右移动
那么取区间(j,i]答案一定等于
s[i]-s[j]-s[x]-s[x-d](x-d<=j&&x<=i)的最小值
使得答案最小的x也是随着i单调递增的,于是使用单调队列优化
代码是简单的,ac是艰难的
#include<stdio.h> #include<stdlib.h> #include<iostream> #include<string> #include<string.h> #include<algorithm> #include<math.h> #include<queue> #include<map> #include<vector> #include<set> #define il inline #define re register using namespace std; typedef long long ll; const int N=2000001; int n,d,a[N],ans,head=1,tail=0,t[N]; ll s[N],p; il ll read(){ ll hs=0;char c=getchar(); while(!isdigit(c)) c=getchar(); while(isdigit(c)){ hs=(hs<<3)+(hs<<1)+c-‘0‘; c=getchar(); } return hs; } int main(){ n=read();p=read();ans=d=read(); for(int i=1;i<=n;i++) a[i]=read(); for(int i=1;i<=n;i++) s[i]=s[i-1]+a[i]; for(int i=d,j=0;i<=n;i++){ while(head<=tail&&s[i]-s[i-d]>s[t[tail]]-s[t[tail]-d]) tail--; t[++tail]=i; while(s[i]-s[j]-s[t[head]]+s[t[head]-d]>p){ j++; if(t[head]-d<j){ head++; } } ans=max(ans,i-j); } printf("%d\n",ans); return 0; }
bzoj4385 & POJ2015 Wilcze do?y
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。