TYVJ 1730 二逼平衡树 线段树套平衡树

没什么好说的。。被坑了。。

查询排名为k的数时，二分这个数，转化成查找w的排名。

若w这个数不存在，就先插入w，然后把w旋转到根，既可以求w的排名。

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <algorithm>

#define INF 100000000
#define N 1110000
#define BUG system("pause")

using namespace std;

int sz[N],fa[N],son[N][2],val[N],root[N];
int n,cnt,m;
int a[N],b[N];
int opt,sq,sum,suc,pre,pos;
int q[N],tot;

struct ST
{
    int l,r;
}T[N<<2];

inline void prt(int x)
{
    if(!x) return;
    prt(son[x][0]);
    printf("%d   ",val[x]);
    prt(son[x][1]);
}

inline void pushup(int x)
{
    if(!x) return;
    sz[x]=sz[son[x][0]]+sz[son[x][1]]+1;
}

inline void link(int x,int y,int c)
{
    fa[x]=y; son[y][c]=x;
}

void rotate(int x,int c)
{
    int y=fa[x];
    link(x,fa[y],son[fa[y]][1]==y);
    link(son[x][!c],y,c);
    link(y,x,!c);
    pushup(y);
}

void splay(int x,int g,int &rt)
{
    while(fa[x]!=g)
    {
        int y=fa[x];
        int cy=(son[fa[y]][1]==y),cx=(son[y][1]==x);
        if(fa[y]==g) rotate(x,cx);
        else
        {
            if(cx==cy) rotate(y,cy);
            else rotate(x,cx);
            rotate(x,cy);
        }    
    }
    pushup(x);
    if(!g) rt=x;
}

inline int getnum()
{
    if(tot) return q[tot--];
    return ++cnt;
}

inline void newnode(int y,int &x,int sp)
{
    x=getnum();
    fa[x]=y; val[x]=sp; sz[x]=1;
    son[x][1]=son[x][0]=0;
}

inline void pack(int &u,int l,int r,int f)
{
    if(r<l) return;
    int mid=(l+r)>>1;
    newnode(f,u,a[mid]);
    pack(son[u][0],l,mid-1,u);
    pack(son[u][1],mid+1,r,u);
    pushup(u);
}

inline void bsplay(int u,int l,int r)
{
    newnode(0,root[u],-INF);
    newnode(root[u],son[root[u]][1],INF);
    sz[root[u]]=2;
    pack(son[son[root[u]][1]][0],l,r,son[root[u]][1]);
    pushup(son[son[root[u]][1]][0]); pushup(son[root[u]][1]);
}

inline void build(int u,int l,int r)
{
    T[u].l=l; T[u].r=r;
    if(l==r)
    {
        bsplay(u,l,r);
        return;
    }
    int mid=(l+r)>>1;
    build(u<<1,l,mid); 
    build(u<<1|1,mid+1,r);
    sort(a+l,a+r+1);
    bsplay(u,l,r);
}

inline void read()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]),b[i]=a[i];
    build(1,1,n);
}

inline int find(int sp,int x)
{
    while(x)
    {
        if(val[x]==sp) return x;
        x=son[x][val[x]<sp];
    }
}

inline int getmin(int x)
{
    while(son[x][0]) x=son[x][0];
    return x;
}

inline int getmax(int x)
{
    while(son[x][1]) x=son[x][1];
    return x;
}

inline void insert(int sp,int &rt)
{
    int x=rt;
    while(son[x][val[x]<sp]) x=son[x][val[x]<sp];
    newnode(x,son[x][val[x]<sp],sp);
    splay(son[x][val[x]<sp],0,rt);
}

inline void del(int sp,int &rt)
{
    int x=find(sp,rt);
    splay(x,0,rt);
    int y=getmax(son[x][0]),z=getmin(son[x][1]);
    splay(y,0,rt); splay(z,y,rt);
    q[++tot]=son[z][0]; son[z][0]=0;
    pushup(z); pushup(y);
}

inline int pred(int x,int sp)
{
    int re=-INF;
    while(x)
    {
        if(val[x]<sp) re=max(val[x],re);
        x=son[x][val[x]<sp];
    }
    return re;
}

inline int succ(int x,int sp)
{
    int re=INF;
    while(x)
    {
        if(val[x]>sp) re=min(val[x],re);
        x=son[x][val[x]<=sp];//!!!!!
    }
    return re;
}

inline void query(int u,int l,int r)
{
    if(T[u].l>=l&&T[u].r<=r)
    {
        if(opt==1)
        {
            insert(sq,root[u]);
            sum+=sz[son[root[u]][0]]-1;
            del(sq,root[u]);
        }
        else if(opt==3)
        {
            del(b[pos],root[u]);
            insert(sq,root[u]);
        }
        else if(opt==4)
        {
            pre=max(pre,pred(root[u],sq));
        }
        else
        {
            suc=min(suc,succ(root[u],sq));
        }
        return;
    }
    int mid=(T[u].l+T[u].r)>>1;
    if(r<=mid) query(u<<1,l,r);
    else if(l>mid) query(u<<1|1,l,r);
    else query(u<<1,l,mid),query(u<<1|1,mid+1,r);
    if(opt==3)
    {
        del(b[pos],root[u]);
        insert(sq,root[u]);
    }
}

inline void go()
{
    for(int i=1,aa,bb,k;i<=m;i++)
    {
        scanf("%d",&opt);
        if(opt==1)
        {
            scanf("%d%d%d",&aa,&bb,&sq);
            sum=0;
            query(1,aa,bb);
            printf("%d\n",sum+1);
        }
        else if(opt==2)
        {
            scanf("%d%d%d",&aa,&bb,&k);
            opt=1;
            int l=0,r=INF,ans;
            while(l<=r)
            {
                int mid=(l+r)>>1;
                sum=0;
                sq=mid; query(1,aa,bb);
                if(sum+1<=k) ans=mid,l=mid+1;
                else r=mid-1;
            }
            printf("%d\n",ans);
        }
        else if(opt==3)
        {
            scanf("%d%d",&pos,&sq);
            query(1,pos,pos);
            b[pos]=sq;
        }
        else if(opt==4)
        {
            scanf("%d%d%d",&aa,&bb,&sq);
            pre=-INF;
            query(1,aa,bb);
            printf("%d\n",pre);
        }
        else
        {
            scanf("%d%d%d",&aa,&bb,&sq);
            suc=INF;
            query(1,aa,bb);
            printf("%d\n",suc);
        }
    }
}

int main()
{
    read(),go();
    return 0;
}