BZOJ 3110, K大数查询

2019-11-11 01:59:32

字体：大中小

来源：转载

供稿：网友

PRoblem

传送门

Mean

要求维护一个数列，支持在某部分的每个位置填上一个数以及查询某部分第K大的数字。

Analysis

终于学习了一下树套树的姿势……其实和自己想的也差不多……但是不学还是打不出代码来…… 外层是权值线段树，内层是区间线段树。要注意的细节比较多。二分查询过程中结果可能超过 Maxlongint ，需要用 unsigned int 类型传递；权值可以翻转后整体右移等等。但最主要的还是实现树套树啦~~~

Code

#include<cstdio>const int N=50000,L=100001,M=16000005;int n,m,cmd,a,b,c,cnt,root[M],ls[M],rs[M];unsigned int sum[M],tag[M];void read(int &x){ char c; bool p=0; while((c=getchar())<'0' || c>'9') if(c=='-') p=1; x=c-'0'; while((c=getchar())>='0' && c<='9') x=x*10+c-'0'; if(p) x=-x;}void down(int o,int l,int r){ if(!ls[o]) ls[o]=++cnt; if(!rs[o]) rs[o]=++cnt; int mid=l+r>>1; tag[ls[o]]+=tag[o],tag[rs[o]]+=tag[o]; sum[ls[o]]+=(mid-l+1)*tag[o],sum[rs[o]]+=(r-mid)*tag[o]; tag[o]=0;}void modify(int &o,int l,int r,int ql,int qr){ if(!o) o=++cnt; if(tag[o] && l<r) down(o,l,r); if(ql==l && r==qr){ sum[o]+=r-l+1; tag[o]++; return; } int mid=l+r>>1; if(qr<=mid) modify(ls[o],l,mid,ql,qr); else if(ql>mid) modify(rs[o],mid+1,r,ql,qr); else{ modify(ls[o],l,mid,ql,mid); modify(rs[o],mid+1,r,mid+1,qr); } sum[o]=sum[ls[o]]+sum[rs[o]];}void insert(int o,int l,int r,int a,int b,int q){ while(l<r){ int mid=l+r>>1; modify(root[o],1,n,a,b); if(q<=mid) r=mid,o<<=1; else l=mid+1,o=o<<1|1; } modify(root[o],1,n,a,b);}unsigned int find(int o,int l,int r,int ql,int qr){ if(!o) return 0; if(l==ql && r==qr) return sum[o]; if(tag[o]) down(o,l,r); int mid=l+r>>1; if(qr<=mid) return find(ls[o],l,mid,ql,qr); if(ql>mid) return find(rs[o],mid+1,r,ql,qr); return find(ls[o],l,mid,ql,mid)+find(rs[o],mid+1,r,mid+1,qr);}int query(int o,int l,int r,int a,int b,unsigned int c){ while(l<r){ int mid=l+r>>1; unsigned int t=find(root[o<<1],1,n,a,b); if(t<c) l=mid+1,c-=t,o=o<<1|1; else r=mid,o<<=1; } return l;}int main(){ read(n),read(m); while(m--){ read(cmd),read(a),read(b),read(c); if(cmd==1) insert(1,1,L,a,b,N-c+1); else printf("%d/n",N+1-query(1,1,L,a,b,c)); } return 0;}

上一篇：1021. Deepest Root 解析

下一篇：设计模式初探（二）「Facade模式」

学习交流

索泰发布一款GTX 1070 Mini迷你版本:小机

索泰发布一款GTX 1070 Mini迷你版本:小机箱大爱...

热门图片

猜你喜欢的新闻

猜你喜欢的关注