[BZOJ3065]带插入区间K小值 解题报告 替罪羊树+值域线段树
Posted
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了[BZOJ3065]带插入区间K小值 解题报告 替罪羊树+值域线段树相关的知识,希望对你有一定的参考价值。
刚了一天的题终于切掉了,数据结构题的代码真**难调,这是我做过的第一道树套树题,做完后感觉对树套树都有阴影了......下面写一下做题记录.
Portal Gun:[BZOJ3065]带插入区间k小值.
这道题的题面其实都提醒怎么做了,维护区间k小值用值域线段树,但要维护一个插入操作,树状数组套主席树也用不了,那么这道题还剩下平衡树可以搞,那就上平衡树吧.
我这里的做法,因为要维护序列的顺序,所以我这里用到替罪羊树套值域线段树:我们在替罪羊树的每个节点都套一颗值域线段树,记录以该节点为根的子树的值域的 size ,替罪羊树中我们维护每一个数在数列中的序号,在插入一个数时,就用它的值去更新插入时经过的所有点,就可以完成插入操作.至于修改操作,其实也就是插入操作,新值 size++ ,旧值 size-- .
关于复杂度:对于每次插入,替罪羊树是 O(logn) ,插入值域线段树是 O(logn) ,复杂度为 O(log^2) .对于每次查询,替罪羊树找区间是 O(logn) , 查询值域线段树是 O(logn) ,复杂度为 O(log^2) ,替罪羊树重构复杂度均摊为 O(log^2) .
#include<iostream>
#include<cstdio>
#include<cstdlib>
#define maxn (70005)
#define al 0.75
#define il inline
#define RG register
using namespace std;
il int gi(){ RG int x=0,q=1; RG char ch=getchar(); while( ( ch<‘0‘ || ch>‘9‘ ) && ch!=‘-‘ ) ch=getchar();
if( ch==‘-‘ ) q=-1,ch=getchar(); while(ch>=‘0‘ && ch<=‘9‘) x=x*10+ch-48,ch=getchar(); return q*x; }
struct node{int l,r,size;}t[maxn*200];//值域线段树
int n,m,ans,flag,root,cnt,Max=70000;
int id[maxn],w[maxn],rt[maxn],ls[maxn],rs[maxn]; //替罪羊树
int sta[maxn*100],top; //存回收节点
il int newnode(){ if(!top) return ++cnt; else return sta[top--]; }
il bool BALANCE(int x){
return t[ rt[x] ].size*al>t[ rt[ls[x]] ].size
&& t[ rt[x] ].size*al>t[ rt[rs[x]] ].size;
}
il void recycle(int &x){
if(!x) return ; sta[++top]=x;
recycle(t[x].l),recycle(t[x].r);
t[x].size=0,x=0;
}
il void insert(RG int &x,RG int l,RG int r,RG int val,RG int sum){ //线段树插入
if(!x) x=newnode();
if(l==r){ t[x].size+=sum; return ; }
RG int mid=(l+r)>>1;
if(val<=mid) insert( t[x].l,l,mid,val,sum );
else insert( t[x].r,mid+1,r,val,sum );
t[x].size=t[ t[x].l ].size+t[ t[x].r ].size;
if(!t[x].size) recycle(x);
}
il void build(RG int &x,RG int l,RG int r){
if(l>r) return ;
if(l==r){ x=id[l]; insert(rt[x],0,Max,w[x],1); return ; }
RG int mid=(l+r)>>1; x=id[mid];
build(ls[x],l,mid-1); build(rs[x],mid+1,r);
for(RG int i=l;i<=r;i++) insert(rt[x],0,Max,w[id[i]],1);
}
int cur[maxn*100],c_size;
il void dfs(RG int &x){
if(!x) return ; recycle(rt[x]);
dfs(ls[x]); cur[++c_size]=x; dfs(rs[x]);
x=0;
}
il void REBUILD(int &x){
dfs(x);
for(RG int i=1;i<=c_size;i++) id[i]=cur[i];
build(x,1,c_size); c_size=0;
}
il void INSERT(RG int &x,RG int l,RG int val){
if(!x){
x=++n; w[x]=val;
insert(rt[x],0,Max,val,1);
return ;
}
insert( rt[x],0,Max,val,1 );
int L=t[ rt[ ls[x] ] ].size;
if(l<=L) INSERT(ls[x],l,val);
else INSERT(rs[x],l-L-1,val);
if(BALANCE(x)){ if(flag){
if(ls[x]==flag) REBUILD(ls[x]);
else REBUILD(rs[x]); flag=0;
}
}
else{ flag=x; if(flag==root) REBUILD(root); }
}
il int modify(RG int x,RG int l,RG int val){
insert(rt[x],0,Max,val,1); //插入新值
int tt,L=t[ rt[ls[x]] ].size;
if(l==L+1){ tt=w[x]; w[x]=val; }
else if(l<=L) tt=modify(ls[x],l,val);
else tt=modify(rs[x],l-L-1,val);
insert(rt[x],0,Max,tt,-1); //删除原值
return tt;
}
int tt[maxn*100],t_size,v[maxn*100],v_size;
il void query(int k,int l,int r){
int L=t[ rt[ ls[k] ] ].size,R=t[ rt[k] ].size;
if(l==1 && r==R){ tt[++t_size]=rt[k]; return ; }
if(l<=L+1 && r>=L+1) v[++v_size]=w[k];
if(r<=L) query(ls[k],l,r);
else if(l>L+1) query(rs[k],l-L-1,r-L-1);
else{
if(l<=L) query(ls[k],l,L);
if(R>L+1) query(rs[k],1,r-L-1);
}
}
il int QUERY(int L,int R,int k){
query(root,L,R); k--;
int l=0,r=Max;
while(l<r){
RG int mid=(l+r)>>1,sum=0;
for(RG int i=1;i<=t_size;i++) sum+=t[ t[ tt[i] ].l ].size;
for(RG int i=1;i<=v_size;i++) if(v[i]>=l && v[i]<=mid) sum++;
if(k<sum){
for(RG int i=1;i<=t_size;i++) tt[i]=t[ tt[i] ].l;
r=mid;
}
else{
for(RG int i=1;i<=t_size;i++) tt[i]=t[ tt[i] ].r;
l=mid+1,k-=sum;
}
}
t_size=0,v_size=0;
return l;
}
il void init(){
n=gi(); for(RG int i=1;i<=n;i++) w[i]=gi(),id[i]=i;
build(root,1,n);
}
il void work(){
m=gi(); char s[2]; int x,y,k;
while(m--){
scanf("%s",s);
x=gi()^ans,y=gi()^ans;
if(s[0]==‘M‘) modify(root,x,y);
if(s[0]==‘I‘){ flag=0; INSERT(root,x-1,y); }
if(s[0]==‘Q‘){ k=gi()^ans;ans=QUERY(x,y,k);printf("%d\n",ans); }
}
}
int main(){ init(); work(); return 0; }
以上是关于[BZOJ3065]带插入区间K小值 解题报告 替罪羊树+值域线段树的主要内容,如果未能解决你的问题,请参考以下文章