bzoj 3196/ Tyvj 1730 二逼平衡树 （线段树套平衡树）

#include<cstdio>
#include<algorithm>
using namespace std;
#define M 3000001
#define N 50001

    int id=0;
    int siz[M],cnt[M],key[M];
    int fa[M],ch[M][2];
    struct TREE
    {
        int root,l,r;
    }tr[N<<2];
    int data[N];
    void Splay_up(int k)
    {
        siz[k]=siz[ch[k][0]]+siz[ch[k][1]]+cnt[k];
    }
    void rotate(int x,int &goal)
    {
        int y=fa[x],z=fa[y];
        int l=ch[y][1]==x,r=l^1;
        if(y==goal) goal=x;
        else ch[z][ch[z][1]==y]=x;
        ch[y][l]=ch[x][r]; ch[x][r]=y; 
        fa[x]=z; fa[y]=x; fa[ch[y][l]]=y;
        Splay_up(y);
    }
    void splay(int x,int &goal)
    {
        int y,z;
        while(x!=goal)
        {
            y=fa[x]; z=fa[y];
            if(y!=goal) 
            {
                if(ch[z][0]==y ^ ch[y][0]==x) rotate(x,goal);
                else rotate(y,goal);
            }
            rotate(x,goal);
            Splay_up(x);
        }
    }
    void Splay_insert(int &rt,int w)
    {    
        if(!rt)
        {
            rt=++id;
            siz[rt]=cnt[rt]=1;
            key[rt]=w;
            return;
        }
        if(key[rt]==w) cnt[rt]++;
        else if(w<key[rt]) Splay_insert(ch[rt][0],w),fa[ch[rt][0]]=rt;
        else Splay_insert(ch[rt][1],w),fa[ch[rt][1]]=rt;
        Splay_up(rt);
    }
    int Splay_getpre(int rt,int x)
    {
        int now=rt,ret;
        while(now)
        {
            if(key[now]>=x) now=ch[now][0];
            else ret=now,now=ch[now][1];
        }
        return ret;
    }
    int Splay_getsuc(int rt,int x)
    {
        int now=rt,ret;
        while(now)
        {
            if(key[now]<=x) now=ch[now][1];
            else ret=now,now=ch[now][0];
        }
        return ret;
    }
    int Splay_number(int now,int w)
    {
        while(1)
        {
            if(key[now]==w) return now;
            if(w<key[now]) now=ch[now][0];
            else now=ch[now][1];
        }
    }
    void del(int & rt,int w)
    {
        int num=Splay_number(rt,w);
        splay(num,rt);
        if(cnt[num]>1) cnt[num]--, siz[num]--;
        else
        {
            int pre=Splay_getpre(rt,w);
            splay(pre,rt);
            ch[pre][1]=ch[num][1];
            fa[ch[pre][1]]=pre;
            Splay_up(pre);
        }
    }
    void add_dummy(int rt,int w,int d)
    {
        ch[rt][d]=++id; siz[rt]++;
        fa[id]=rt; siz[id]=cnt[id]=1; key[id]=w;
    }
    
    void build(int k,int l,int r)
    {
        tr[k].l=l; tr[k].r=r;
        if(l==r) return;
        int mid=l+r>>1;
        build(k<<1,l,mid); 
        build(k<<1|1,mid+1,r);    
    }
    void Segment_insert(int k,int pos,int w)
    {
         Splay_insert(tr[k].root,w);
        if(tr[k].l==tr[k].r) return;
        int mid=tr[k].l+tr[k].r>>1;
        if(pos<=mid) Segment_insert(k<<1,pos,w);
        else Segment_insert(k<<1|1,pos,w);
    }
    void dummy(int k)
    {
        int pre=Splay_getsuc(tr[k].root,-1),suc=Splay_getpre(tr[k].root,1e8);
        splay(pre,tr[k].root);
        add_dummy(pre,-1e8-1,0);
        splay(suc,tr[k].root);
        add_dummy(suc,1e8+1,1);
        if(tr[k].l==tr[k].r) return;
        dummy(k<<1); dummy(k<<1|1);
    }
    int order(int k,int opl,int opr,int w)
    {
        if(tr[k].l>=opl && tr[k].r<=opr)
        {
            int pre=Splay_getpre(tr[k].root,w);
            splay(pre,tr[k].root);
            return siz[ch[pre][0]]+cnt[pre]-1;
        }
        int mid=tr[k].l+tr[k].r>>1,tmp=0;
        if(opl<=mid) tmp+=order(k<<1,opl,opr,w);
        if(opr>mid) tmp+=order(k<<1|1,opl,opr,w);
        return tmp;
    }
    int Segment_number(int l,int r,int w)
    {
        int ll=1,rr=1e8,mid,tmp,ans;
        while(ll<=rr)
        {
            mid=ll+rr>>1;
            tmp=order(1,l,r,mid);
            if(tmp<w) ans=mid,ll=mid+1;
            else rr=mid-1;
        }
        return ans;
    }
    void modify(int k,int pos,int before,int now)
    {
        del(tr[k].root,before);
        Splay_insert(tr[k].root,now);
        if(tr[k].l==tr[k].r)
        {
            data[tr[k].l]=now;
            return;
        }
        int mid=tr[k].l+tr[k].r>>1;
        if(pos<=mid) modify(k<<1,pos,before,now);
        else modify(k<<1|1,pos,before,now);
    }
    int Segment_getpre(int k,int l,int r,int w)
    {
        if(tr[k].l>=l && tr[k].r<=r) return key[Splay_getpre(tr[k].root,w)];
        int mid=tr[k].l+tr[k].r>>1,tmp=-1e8-1;
        if(l<=mid) tmp=max(tmp,Segment_getpre(k<<1,l,r,w));
        if(r>mid) tmp=max(tmp,Segment_getpre(k<<1|1,l,r,w));
        return tmp; 
    }
    int Segment_getsuc(int k,int l,int r,int w)
    {
        if(tr[k].l>=l && tr[k].r<=r) return key[Splay_getsuc(tr[k].root,w)];
        int mid=tr[k].l+tr[k].r>>1,tmp=1e8+1; 
        if(l<=mid) tmp=min(tmp,Segment_getsuc(k<<1,l,r,w));
        if(r>mid) tmp=min(tmp,Segment_getsuc(k<<1|1,l,r,w));
        return tmp;
     }
     
int main()
{
    freopen("psh.in","r",stdin);
    freopen("psh.out","w",stdout);
    int n,m;
    scanf("%d%d",&n,&m);
    build(1,1,n);
    for(int i=1;i<=n;i++) 
    {
        scanf("%d",&data[i]); 
        Segment_insert(1,i,data[i]);
    }
    dummy(1);
    int opt,l,r,k;
    while(m--)
    {
        scanf("%d",&opt);
        if(opt==1)
        {
            scanf("%d%d%d",&l,&r,&k);
            printf("%d\n",order(1,l,r,k)+1);
        }
        else if(opt==2)
        {
            scanf("%d%d%d",&l,&r,&k);
            printf("%d\n",Segment_number(l,r,k));
        }
        else if(opt==3)
        {
            scanf("%d%d",&l,&k);
            modify(1,l,data[l],k);
        }
        else if(opt==4)
        {
            scanf("%d%d%d",&l,&r,&k);
            printf("%d\n",Segment_getpre(1,l,r,k));
        }
        else
        {
            scanf("%d%d%d",&l,&r,&k);
            printf("%d\n",Segment_getsuc(1,l,r,k));
        }
    }
}