BZOJ 3196: Tyvj 1730 二逼平衡树 [树套树]

题解不会写,标记一下,抄的是同学的板子

#include<cstdio>
#include<algorithm>
#define N 50010
#define M N*30
using namespace std;

int n,m,sz,ans,a[N],root[M],size[M],fa[M],ch[M][2],v[M];

inline char nc(){
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}

inline int read(){
    int a=0;char c=nc(),f=1;
    while(c<'0'||c>'9'){if(c=='-')f=-1;c=nc();}
    while(c>='0'&&c<='9'){a=a*10+c-'0';c=nc();}
    return a*f;
}

inline int get(int x){return ch[fa[x]][1]==x;}

inline void update(int x){
    if(x){
    size[x]=1;
    if(ch[x][0]) size[x]+=size[ch[x][0]];
    if(ch[x][1]) size[x]+=size[ch[x][1]];
    }
}

inline void rotate(int x){
    int y=fa[x],opt=(ch[y][1]==x);
    ch[y][opt]=ch[x][opt^1];
    if(ch[x][opt^1]) fa[ch[x][opt^1]]=y;
    if(fa[x]=fa[y]) ch[fa[y]][ch[fa[y]][1]==y]=x;
    fa[y]=x,ch[x][opt^1]=y;
    update(y),update(x);
}

inline void splay(int x,int ed,int pos){
    for(int fat=fa[x];fat!=ed;rotate(x),fat=fa[x])
    if(fa[fat]!=ed) rotate((x==ch[fat][0])==(fat==ch[fa[fat]][0])?fat:x);
    if(!ed) root[pos]=x;
}

inline void insert(int rt,int x){
    int nw=root[rt];
    while(ch[nw][v[nw]<x]) nw=ch[nw][v[nw]<x];
    v[++sz]=x,fa[sz]=nw,ch[sz][0]=ch[sz][1]=0;
    if(nw) ch[nw][v[nw]<x]=sz;
    splay(sz,0,rt);
}

inline int srank(int rt,int x){
    int nw=root[rt],ans=0;
    while(nw)
    if(v[nw]<x) ans+=size[ch[nw][0]]+1,nw=ch[nw][1];
    else nw=ch[nw][0];
    return ans;
}

inline void del(int rt,int x){
    int nw=root[rt];
    while(v[nw]!=x) nw=ch[nw][v[nw]<x];
    splay(nw,0,rt);
    if(!ch[root[rt]][0]){
    fa[root[rt]=ch[root[rt]][1]]=0;
    return;
    }
    nw=ch[root[rt]][0];
    while(ch[nw][1]) nw=ch[nw][1];
    splay(nw,root[rt],rt);
    fa[ch[nw][1]=ch[root[rt]][1]]=nw,fa[root[rt]=nw]=0;
    update(nw);
    return;
}

inline void spre(int rt,int x){
    int nw=root[rt];
    while(nw)
    if(v[nw]<x) ans=max(ans,v[nw]),nw=ch[nw][1];
    else nw=ch[nw][0];
}

inline void ssuf(int rt,int x){
    int nw=root[rt];
    while(nw)
    if(x<v[nw]) ans=min(ans,v[nw]),nw=ch[nw][0];
    else nw=ch[nw][1];
}

#define ls (x<<1)
#define rs (x<<1|1)
#define mid (l+r>>1)

void change(int x,int l,int r,int q,int v,int opt){
    if(opt) insert(x,v);
    else del(x,v);
    if(l==r) return;
    if(q<=mid) change(ls,l,mid,q,v,opt);
    else change(rs,mid+1,r,q,v,opt);
}

inline int rank(int x,int l,int r,int ql,int qr,int v){
    if(l>qr||r<ql) return 0;
    if(l>=ql&&r<=qr) return srank(x,v);
    return rank(ls,l,mid,ql,qr,v)+rank(rs,mid+1,r,ql,qr,v);
}

inline void pre(int x,int l,int r,int ql,int qr,int v){
    if(l>qr||r<ql) return;
    if(l>=ql&&r<=qr) return spre(x,v);
    pre(ls,l,mid,ql,qr,v),pre(rs,mid+1,r,ql,qr,v);
}

inline void suf(int x,int l,int r,int ql,int qr,int v){
    if(l>qr||r<ql) return;
    if(l>=ql&&r<=qr) return ssuf(x,v);
    suf(ls,l,mid,ql,qr,v),suf(rs,mid+1,r,ql,qr,v);
}

inline long long find(int x,int y,int k){
    long long l=0,r=2e9;
    while(l<=r)
    if(rank(1,1,n,x,y,mid)+1<=k) l=mid+1;
    else r=mid-1;
    return r;
}

int main(){
    n=read(),m=read();
    for(int i=1;i<=n;++i) a[i]=read(),change(1,1,n,i,a[i],1);
    while(m--){
    int opt=read(),l,r,k,pos;
    switch(opt){
    case 1:l=read(),r=read(),k=read();printf("%d\n",rank(1,1,n,l,r,k)+1);break;
    case 2:l=read(),r=read(),k=read();printf("%d\n",find(l,r,k));break;
    case 3:pos=read(),k=read();change(1,1,n,pos,a[pos],0),change(1,1,n,pos,a[pos]=k,1);break;
    case 4:l=read(),r=read(),k=read();ans=-(1ll<<31)+1,pre(1,1,n,l,r,k);printf("%d\n",ans);break;
    case 5:l=read(),r=read(),k=read();ans=(1ll<<31)-1,suf(1,1,n,l,r,k);printf("%d\n",ans);break;
    }
    }
    return 0;
}