BZOJ1861 [Zjoi2006]Book 书架

从6点调到了现在22:19:07。。脑子晕倒死。

用splay做，一开始怎么想也不知道该怎么play。

想了个办法，用pos[i]表示编号为i的书在树上的节点编号

s[i]表示树上的节点i代表的是哪本书。

val[i]表示节点i的权值，这里的权值按照书从上到下的大小顺序来赋值，用来建树

TOP：先把树里面代表书S的那个节点删掉，然后再重新建一个节点把S放进去，这时候，

　　　　为了使新的节点在最前方，开一个mmin表示已经出现过的最小的数，

　　　　每次要把节点加入进去的时候插入几点的权值就是 --mmin；

BOT：同上；

INSERT：找到S的前驱或后继，然后调换一下顺序就ok

ASK：查询排名

QUERY：查询排名为K的书的编号。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 200005;
int fa[maxn],son[2][maxn],val[maxn],pos[maxn],siz[maxn],s[maxn];
int root,tot,mmax,mmin;
void rota(int w,int x){
    int y = fa[x];
    son[!w][y] = son[w][x];
    if(son[w][x])fa[son[w][x]] = y;
    fa[x] = fa[y];
    if(fa[y])son[y==son[1][fa[y]]][fa[y]] = x;
    fa[y] = x;
    son[w][x] = y;
    siz[y] = siz[y]-siz[x]+siz[son[!w][y]];
    siz[x] = siz[x]-siz[son[!w][y]]+siz[y];
}
void splay(int x,int y){
    while(fa[x]!=y){
        if(fa[fa[x]]==y)rota(x==son[0][fa[x]],x);
        else {
            int w = fa[x]==son[0][fa[fa[x]]];
            if(x==son[w][fa[x]]){
                rota(!w,x);rota(w,x);
            }
            else {
                rota(w,fa[x]);rota(w,x);
            }
        }
    }
    if(y==0)root = x;
}
void Ins(int v){
    int x = root;
    while(1){
        siz[x]++;
        if(v<val[x]){
            if(son[0][x])x = son[0][x];
            else break;
        }
        else {
            if(son[1][x])x = son[1][x];
            else break;
        }
    }
    fa[++tot] = x;
    val[tot] = v;
    siz[tot] = 1;
    if(v<val[x])son[0][x] = tot;
    else son[1][x] = tot;
    splay(tot,0);
}
int getPre(int x){
    splay(x,0);x = son[0][x];///先把x从根节点拿到子树上
    if(!x)return 0;
    while(son[1][x])x = son[1][x];
    return x;
}
int getNext(int x){
    splay(x,0);x = son[1][x];
    if(!x)return 0;
    while(son[0][x])x = son[0][x];
    return x;
}
void Del(int x){
    splay(x,0);
    int y = getPre(x),z = getNext(x);
    if(y==0&&z==0){root = 0;return;}
    if(y)splay(y,0);
    if(z)splay(z,y);
    if(y==0){son[0][z] = 0,siz[z]--;return;}
    if(z==0){son[1][y] = 0,siz[y]--;return;}
    son[0][z] = 0;
    siz[z]--,siz[y]--;
}
int getRank(int x){
    splay(x,0);
    return siz[son[0][x]];
}
int getK(int k){
    int x = root;
    while(k!=siz[son[0][x]]+1){
        if(k<=siz[son[0][x]])x = son[0][x];
        else {
            k-=(siz[son[0][x]]+1);///以后千万记得先更新k之后再更新x!!!!
            x = son[1][x];
        }
    }
    return s[x];
}
int main()
{
    int n,m;scanf("%d%d",&n,&m);
    for(int i = 1;i<=n;++i){
        int t;scanf("%d",&t);
        if(i==1){
            root = ++tot;
            val[1] = 1;
            siz[1] = 1;
            s[1] = t;
        }
        else Ins(i);
        pos[t] = tot;
        s[tot] = t;
    }
    mmin = 1;mmax = n;
    while(m--){
        char ss[10];
        int a,b,x;
        scanf("%s%d",ss,&a);
        if(ss[0]=='T'){
            Del(pos[a]);
            Ins(--mmin);pos[a] = tot;
            s[tot] = a;
        }
        else if(ss[0]=='B'){
            Del(pos[a]);Ins(++mmax);
            pos[a] = tot;
            s[tot] = a;
        }
        else if(ss[0]=='I'){
            scanf("%d",&b);
            if(!b)continue;
            if(b==-1)x = getPre(pos[a]);
            else x = getNext(pos[a]);
            x = s[x];
            swap(pos[a],pos[x]);
            swap(s[pos[a]],s[pos[x]]);
        }
        else if(ss[0]=='A')printf("%d\n",getRank(pos[a]));
        else printf("%d\n",getK(a));
    }
    return 0;
}