首页 > 代码库 > HDU 5289 Assignment(单调队列)
HDU 5289 Assignment(单调队列)
题意:给T足数据,然后每组一个n和k,表示n个数,k表示最大同意的能力差,接下来n个数表示n个人的能力,求能力差在k之内的区间有几个
分析:维护一个区间的最大值和最小值,使得他们的差小于k,于是採用单调队列
普通单调队列做法:
#include<cstdio> #include<cstring> #include<algorithm> #include<cmath> using namespace std; const int maxn = 1e6+5; int a[maxn]; struct node{ int index; int v; }qd[maxn]; node qx[maxn]; int main() { int t; scanf("%d",&t); while(t--) { int n,k; scanf("%d%d",&n,&k); for(int i=1;i<=n;i++){ scanf("%d",&a[i]); } int st1,st2,ed1,ed2; st1=st2=ed1=ed2=1; long long sum=0; int j=1; for(int i=1;i<=n&&j<=n;i++){ if(i==1){ qd[1].index=qx[1].index=1; qd[1].v=qx[1].v=a[1]; } else{ while(st1<=ed1){ //单调队列维护最大值 if(qd[ed1].v<=a[i]) ed1--; //比a[i]小的并且下标比i小的出队列 else break; } qd[++ed1].v=a[i]; //a[i]入队列 qd[ed1].index=i; while(st2<=ed2){ //单调队列维护最小值 if(qx[ed2].v>=a[i]) ed2--; //比a[i]大并且下标比i小的出队列 else break; } qx[++ed2].v=a[i]; //a[i]入队列 qx[ed2].index=i; while(qd[st1].v-qx[st2].v>=k&&st1<=ed1&&st2<=ed2) //计数 { if(qd[st1].index==j) st1++; if(qx[st2].index==j) st2++; sum+=(i-j); j++; } } } while(j<=n) { sum+=(n-j+1); j++; } printf("%I64d\n",sum); } }
二分单调队列做法:
#include<cstdio> #include<cstring> #include<algorithm> #include<cmath> using namespace std; const int maxn = 1e6+5; int a[maxn]; struct node{ int index; int v; }qd[maxn]; node qx[maxn]; int maxc(int l,int r,int d){ //二分找出d入队列的为止 while(l<=r){ int mid=(l+r)/2; if(qd[mid].v==d) return mid; else if(qd[mid].v>d) l=mid+1; else r=mid-1; } return l; } int minc(int l,int r,int d){ //二分找出d入队列的为止 while(l<=r){ int mid=(l+r)/2; if(qx[mid].v==d) return mid; else if(qx[mid].v<d) l=mid+1; else r=mid-1; } return l; } int main() { int t; scanf("%d",&t); while(t--) { int n,k; scanf("%d%d",&n,&k); for(int i=1;i<=n;i++){ scanf("%d",&a[i]); } int st1,st2,ed1,ed2; st1=st2=ed1=ed2=1; long long sum=0; int j=1; for(int i=1;i<=n&&j<=n;i++){ if(i==1){ qd[1].index=1; qd[1].v=a[1]; qx[1].index=1; qx[1].v=a[1]; } else{ ed1=maxc(st1,ed1,a[i]); //二分找出d入队列的为止,维护最大值 qd[ed1].v=a[i]; qd[ed1].index=i; ed2=minc(st2,ed2,a[i]); //二分找出d入队列的为止,维护最小值 qx[ed2].v=a[i]; qx[ed2].index=i; while(qd[st1].v-qx[st2].v>=k&&st1<=ed1&&st2<=ed2)//计数 { if(qd[st1].index==j) st1++; if(qx[st2].index==j) st2++; sum+=(i-j); j++; } } } while(j<=n) { sum+=(n-j+1); j++; } printf("%I64d\n",sum); } }
HDU 5289 Assignment(单调队列)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。