[bzoj1861][Zjoi2006]Book书架

我写的非常shi，还巨tm慢，不过。。。很好理解qwq

题意：给编号为1~n的书，支持操作：

1.把编号为x的放在第一个

2.放在最后一个

3.上移一个或下移一个或不动

4.询问编号为x的在第几个

5.第x个编号是几

我的方法是一直给每个点一个权值，以权值为关键字维护splay，这样和普通平衡树一样了。所以说好像可以set水过？

要记录编号为x的权值是几。查询直接查，移动时候先删除，再把权值修改成合适的值（第一个就修改成最小值-1，往前一个就修改成前驱和前驱的前驱中间值，类似）。

顺便每个节点维护一下编号就好了。。

#include<bits/stdc++.h>
#define eps 1e-10
using namespace std;
const int N=200010;
typedef double ff;
inline int read(){
    int r=0,f=1,c=getchar();
    while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9'){r=r*10+c-'0';c=getchar();}
    return r*f;
}
int ch[N][2],siz[N],fa[N],f[N];
ff pos[N],w[N];
int rt,tot,now;
void pp(int x){
    siz[x]=siz[ch[x][0]]+siz[ch[x][1]]+1;
}
int get(int x){
    return x==ch[fa[x]][1];
}
void rotate(int x){
    int y=fa[x],z=fa[y],px=get(x),py=get(y);
    int t=ch[x][px^1];
    ch[y][px]=t;fa[t]=y;
    ch[x][px^1]=y;fa[y]=x;
    ch[z][py]=x;fa[x]=z;
    pp(y);
}
void splay(int x){
    int y=fa[x];
    while(y){
        if(fa[y])
        rotate(get(x)==get(y)?y:x);
        rotate(x);y=fa[x];
    }
    pp(x);rt=x;
}
ff minn(){
    int x=rt;
    while(ch[x][0])x=ch[x][0];
    return w[x];
}
ff maxx(){
    int x=rt;
    while(ch[x][1])x=ch[x][1];
    return w[x];
}
int pre(){
    int x=ch[rt][0];
    while(ch[x][1])x=ch[x][1];
    return x;
}
int nxt(){
    int x=ch[rt][1];
    while(ch[x][0])x=ch[x][0];
    return x;
}
int qnum(int rk){
    int x=rt;
    while(1){
        int y=ch[x][0];
        if(y&&siz[y]>=rk)x=y;
        else{
            rk-=(y?siz[y]:0)+1;
            if(rk<=0){
                splay(x);return f[x];
            }
            x=ch[x][1];
        }
    }
}
int qrank(ff val){
    int x=rt;
    while(abs(w[x]-val)>eps)
    x=ch[x][val>w[x]];
    splay(x);
    return siz[ch[x][0]];
}
void ins(ff val){
    if(!rt){
        rt=++tot;w[tot]=val;
        siz[tot]=1;f[tot]=now;
        return;
    }
    int x=rt;
    while(1){
        int y=ch[x][0];
        y=x,x=ch[x][val>w[x]];
        if(!x){
            w[++tot]=val;
            ch[y][val>w[y]]=tot;fa[tot]=y;
            siz[tot]=1;f[tot]=now;splay(tot);
            return;
        }
    }
}
void del(ff val){
    qrank(val);
    if(!ch[rt][0]&&!ch[rt][1]){
        rt=0;return;
    }
    int x=-1;
    x=ch[rt][0]?x:1;
    x=ch[rt][1]?x:0;
    if(~x){
        rt=ch[rt][x];fa[rt]=0;
        return;
    }
    x=ch[rt][1];
    splay(pre());
    ch[rt][1]=x,fa[x]=rt;
    pp(rt);
}
void dfs(int x){
    if(!x)return;
    dfs(ch[x][0]);
    printf("%.0f ",w[x]);
    dfs(ch[x][1]);
}
int main(){
    int n=read(),m=read();
    for(int i=1;i<=n;i++)
    now=read(),ins(i*1000),pos[now]=i*1000;
    while(m--){
        char s[10];scanf("%s",s);
        if(s[0]=='T'){
            int x=read(),y=minn();
            del(pos[x]);
            now=x;ins(y-1000);
            pos[x]=y-1000;
        }
        else if(s[0]=='B'){
            int x=read(),y=maxx();
            del(pos[x]);
            now=x;ins(y+1000);
            pos[x]=y+1000;
        }
        else if(s[0]=='I'){
            int x=read(),o=read();
            if(!o)continue;
            ff t=pos[x],v;int y,z;
            if(~o){
                qrank(t);int y=pre();
                if(abs(w[y]-minn())>eps){
                    splay(y);z=pre();
                    v=(w[y]+w[z])/2;
                }
                else v=w[y]-1000;
                pos[x]=v;now=x;ins(v);
            }
            else{
                qrank(t);int y=nxt();
                if(abs(w[y]-maxx())>eps){
                    splay(y);z=nxt();
                    v=(w[y]+w[z])/2;
                }
                else v=w[y]+1000;
                pos[x]=v;now=x;ins(v);
            }
        }
        else if(s[0]=='A'){
            printf("%d\n",qrank(pos[read()]));
        }
        else{
            printf("%d\n",qnum(read()));
        }
    }
}
/*
10 10 
1 3 2 7 5 8 10 4 9 6
Query 3 
Top 5 
Ask 6 
Bottom 3 
Ask 3 
Top 6 
Insert 4 -1 
Query 5 
Query 2
Ask 2
*/

想了想真的可以set水掉= =