首页 > 学院 > 开发设计 > 正文

Cogs 1345. [ZJOI2013] K大数查询(树套树)

2019-11-14 09:14:25
字体:
来源:转载
供稿:网友
[ZJOI2013] K大数查询 这里写图片描述 /*树套树写法.bzoj过不了.可能有负数要离散吧.线段树套线段树.外层权值线段树,内层区间线段树维护标记.对权值建一棵权值线段树.某个点表示权值在某个范围内的数的个数.然后对每个点建一棵区间线段树.表示该权值范围在某个区间的数的个数.然后查找用类似二分的思想.*/#include<iostream>#include<cstdio>#define MAXN 50001using namespace std;struct data{int lc,rc,sum,size,bj;}tree[MAXN*400];int n,m,root[MAXN],cut;int read(){ int x=0,f=1;char ch=getchar(); while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} while(ch>='0'&&ch<='9') x=x*10+ch-48,ch=getchar(); return x*f;}void push(int k,int l,int r){ if(!tree[k].lc) tree[k].lc=++cut; if(!tree[k].rc) tree[k].rc=++cut; tree[tree[k].lc].bj+=tree[k].bj; tree[tree[k].rc].bj+=tree[k].bj; int mid=(l+r)>>1; tree[tree[k].lc].sum+=(mid-l+1)*tree[k].bj; tree[tree[k].rc].sum+=(r-mid)*tree[k].bj; tree[k].bj=0; return ;}void add(int &k,int l,int r,int x,int y){ if(!k) k=++cut; if(x==l&&r==y) { tree[k].bj++; tree[k].sum+=r-l+1; return ; } if(tree[k].bj) push(k,l,r); int mid=(l+r)>>1; if(y<=mid) add(tree[k].lc,l,mid,x,y); else if(x>mid) add(tree[k].rc,mid+1,r,x,y); else add(tree[k].lc,l,mid,x,mid),add(tree[k].rc,mid+1,r,mid+1,y); tree[k].sum=tree[tree[k].lc].sum+tree[tree[k].rc].sum; return ;}void sloveadd(int x,int y,int z)//先找到它所在权值的位置然后在[x,y]处加入贡献.{ int l=1,r=n,k=1,mid; while(l!=r) { mid=(l+r)>>1; add(root[k],1,n,x,y); if(z<=mid) r=mid,k=k<<1; else l=mid+1,k=(k<<1)+1; } add(root[k],1,n,x,y); return ;}int query(int k,int l,int r,int x,int y){ if(!k) return 0; if(tree[k].bj) push(k,l,r); if(x<=l&&r<=y) return tree[k].sum; int tot=0,mid=(l+r)>>1; if(x<=mid) tot+=query(tree[k].lc,l,mid,x,y); if(y>mid) tot+=query(tree[k].rc,mid+1,r,x,y); return tot;}int slovequery(int x,int y,int z){ int l=1,r=n,k=1; while(l!=r) { int mid=(l+r)>>1; int t=query(root[k<<1],1,n,x,y); if(t>=z) r=mid,k<<=1; else l=mid+1,k=(k<<1)+1,z-=t; } return l;}int main(){ freopen("zjoi13_sequence.in","r",stdin); freopen("zjoi13_sequence.out","w",stdout); int k,x,y,z; n=read(),m=read(); while(m--) { k=read(),x=read(),y=read(),z=read(); if(k&1) sloveadd(x,y,n-z+1); else PRintf("%d/n",n-slovequery(x,y,z)+1); } return 0;}
发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表