首页 > 代码库 > HDU 4417 Super Mario
HDU 4417 Super Mario
题意:
有n个数字 m个查询 每个查询询问区间[l,r]内有几个数字小于h
思路:
由于n个数字是不变的 所以可以利用划分树在nlogn复杂度内求出任意区间的第k小值(这里指建树)
每次查询可以二分查找该区间第几小的数小于等于h 那么答案就是二分出的个数
总复杂度 nlogn + m(logn)^2 即 建树复杂度 + m次询问*每次查询logn*二分复杂度logn
n和m都只有10^5 完全可过
代码:
#include<cstdio> #include<cstring> #include<algorithm> using namespace std; #define N 100010 #define M 20 int tree[M][N],toleft[M][N],sorted[N]; int n,m,t,T; void build(int l,int r,int dep) { if(l==r) return ; int i,mid=(l+r)>>1; int y=sorted[mid],same=mid-l+1,lpos=l,rpos=mid+1; for(i=l;i<=r;i++) { if(tree[dep][i]<y) same--; } for(i=l;i<=r;i++) { if(tree[dep][i]<y) tree[dep+1][lpos++]=tree[dep][i]; else if(tree[dep][i]==y&&same>0) { tree[dep+1][lpos++]=tree[dep][i]; same--; } else tree[dep+1][rpos++]=tree[dep][i]; toleft[dep][i]=toleft[dep][l-1]+lpos-l; } build(l,mid,dep+1); build(mid+1,r,dep+1); } int query(int L,int R,int l,int r,int dep,int k) { if(l==r) return tree[dep][l]; int mid=(L+R)>>1,amt=toleft[dep][r]-toleft[dep][l-1]; if(amt>=k) { int fl=L+toleft[dep][l-1]-toleft[dep][L-1]; int fr=fl+amt-1; return query(L,mid,fl,fr,dep+1,k); } else { int fr=r+toleft[dep][R]-toleft[dep][r]; int fl=fr-(r-l-amt); return query(mid+1,R,fl,fr,dep+1,k-amt); } } int main() { int i,j,l,r,h,fr,ed,md,tmp,ans; while(~scanf("%d",&T)) { for(t=1;t<=T;t++) { printf("Case %d:\n",t); scanf("%d%d",&n,&m); for(i=1;i<=n;i++) { scanf("%d",&sorted[i]); tree[0][i]=sorted[i]; } sort(sorted+1,sorted+1+n); build(1,n,0); while(m--) { scanf("%d%d%d",&l,&r,&h); l++; r++; fr=1; ed=r-l+1; ans=0; while(fr<=ed) { md=(fr+ed)>>1; tmp=query(1,n,l,r,0,md); if(tmp<=h) { fr=md+1; ans=md; } else ed=md-1; } printf("%d\n",ans); } } } return 0; }
HDU 4417 Super Mario
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。