首页 > 代码库 > HDU 4417 Super Mario 划分树/树状数组
HDU 4417 Super Mario 划分树/树状数组
题目大意:给定一个序列,求区间内小于等于某数的元素数量
首先区间排名一看就是划分树 不过和第k小不一样 我们需要做一些处理
第一种处理方式是二分答案 然后转换成区间第k小 好方法我喜欢 但是这里说的不是这种方法
首先建树,然后对于每个询问,我们讨论k与a[mid]的关系
若k<a[mid],则右子树一定没有小于等于k的数,我们进入左子树查找
若k>=a[mid],则左子树内一定所有数都小于等于k,于是我们将查询区间中进入左子树的元素的数量记入ans,然后查找右区间
递归退出条件是查询区间为空或节点为叶节点
#include<cstdio> #include<cstring> #include<iostream> #include<algorithm> #define M 100100 using namespace std; int n,m,cnt,a[M],b[M],c[M]; int s[20][M]; void Build_Tree(int l,int r,int dpt) { int i,mid=l+r>>1; int l1=l,l2=mid+1; int left=mid-l+1; if(l==r) return ; for(i=l;i<=r;i++) left-=(a[i]<c[mid]); for(i=l;i<=r;i++) { if(a[i]>c[mid]||a[i]==c[mid]&&!left) b[l2++]=a[i],s[dpt][i]=(i==l?0:s[dpt][i-1] ); else b[l1++]=a[i],s[dpt][i]=(i==l?1:s[dpt][i-1]+1),left-=(a[i]==c[mid]); } memcpy( a+l , b+l , sizeof(a[0])*(r-l+1) ); Build_Tree(l,mid,dpt+1); Build_Tree(mid+1,r,dpt+1); } int Get_Ans(int l,int r,int dpt,int x,int y,int k) { int mid=l+r>>1; int l1=(x==l?0:s[dpt][x-1]),l2=s[dpt][y]; if(x>y) return 0; if(l==r) return a[mid]<=k; if(k<c[mid]) return Get_Ans(l,mid,dpt+1,l+l1,l+l2-1,k); else return l2-l1+Get_Ans(mid+1,r,dpt+1,(mid+1)+(x-l-l1),(mid+1)+(y-l+1-l2)-1,k); } int main() { int T,i,x,y,k; for(cin>>T;T;T--) { printf("Case %d:\n",++cnt); cin>>n>>m; for(i=1;i<=n;i++) scanf("%d",&a[i]),c[i]=a[i]; sort(c+1,c+n+1); Build_Tree(1,n,0); for(i=1;i<=m;i++) scanf("%d%d%d",&x,&y,&k),printf("%d\n", Get_Ans(1,n,0,x+1,y+1,k) ); } }
后来看了看题解发现这题树状数组也能做。。。离线思想,先OTZ一下
我们把询问按照询问的值排序,对于每一个值,我们把小于等于这个值的元素位置都插进树状数组,然后查询(x-1,y]区间内有多少个元素即可
写完是140MS 划分树是109MS 没办法两次排序的常数有点大0.0
把反复memset改成时间标记的话是125MS 能快一些
然后别忘了输出Case 1...因为这点P事还贡献了一次WA
#include<cstdio> #include<cstring> #include<iostream> #include<algorithm> #define M 100100 using namespace std; struct abcd{ int x,y,num,pos; bool operator < (const abcd &x)const { return num < x.num; } }q[M]; pair<int,int>a[M]; int n,m,cnt,ans[M]; int c[M],tim[M]; void update(int x) { for(;x<=n;x+=x&-x) { if(tim[x]!=cnt) c[x]=0; tim[x]=cnt; c[x]++; } } int getans(int x) { int re=0; for(;x;x-=x&-x) if(tim[x]==cnt) re+=c[x]; return re; } int main() { int T,i,j,x,y,k; for(cin>>T;T;T--) { printf("Case %d:\n",++cnt); cin>>n>>m; for(i=1;i<=n;i++) scanf("%d",&a[i].first),a[i].second=i; sort(a+1,a+n+1); for(i=1;i<=m;i++) scanf("%d%d%d",&q[i].x,&q[i].y,&q[i].num),q[i].pos=i; sort(q+1,q+m+1); j=1; for(i=1;i<=m;i++) { for(;j<=n&&a[j].first<=q[i].num;j++) update(a[j].second); ans[q[i].pos]=getans(q[i].y+1)-getans(q[i].x); } for(i=1;i<=m;i++) printf("%d\n",ans[i]); } }
HDU 4417 Super Mario 划分树/树状数组
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。