BZOJ1861: [Zjoi2006]Book 书架

题目链接

1 2 3 ：删除点，再插入到指定位置。
4 ：求指定点排名
5 ：求第k小

注意编号和位置。。

【代码】

#include <cstdio>
#include <iostream>
#include <queue>
#include <vector>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <stack>
#define N 80005 
#define M 200005
#define INF 1e9+1
using namespace std;
typedef long long ll;
typedef pair<ll,ll> pa;

int read()
{
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*f;
}

int n,rt,cnt,m;
int son[N][2],fa[N],sz[N],pos[N],num[N];

void pushup(int k)
{
    int l=son[k][0],r=son[k][1];
    sz[k]=sz[l]+sz[r]+1;
}

void Build(int l,int r,int f)
{
    if(l>r) return;
    if(l==r)
    {
        sz[l]=1;fa[l]=f;
        son[f][l>f]=l;return;
    }
    int mid=l+r>>1;
    Build(l,mid-1,mid);Build(mid+1,r,mid);
    fa[mid]=f;son[f][mid>f]=mid;pushup(mid);
}

void Rotate(int x,int &k)
{
    int y=fa[x],z=fa[y],l,r;
    l=son[y][1]==x;r=l^1;
    if(y==k) k=x;
    else son[z][son[z][1]==y]=x;
    fa[x]=z;fa[y]=x;fa[son[x][r]]=y;
    son[y][l]=son[x][r];son[x][r]=y;
    pushup(y);pushup(x);
}

void Splay(int x,int &k)
{
    while(x!=k)
    {
        int y=fa[x],z=fa[y];
        if(y!=k) {
            if(son[z][0]==y^son[y][0]==x) Rotate(x,k);
            else Rotate(y,k);
        }
        Rotate(x,k);
    }
}

int Find(int k,int x)
{
    int l=son[k][0],r=son[k][1];
    if(sz[l]+1==x) return k;
    if(sz[l]+1>x) return Find(l,x);
    return Find(r,x-sz[l]-1);
}

int Ask(int x)
{
    Splay(x,rt);
    printf("%d\n",sz[son[x][0]]-1);
}

void Del(int k)
{
    int x,y,z;
    x=Find(rt,k-1);y=Find(rt,k+1);
    Splay(x,rt);Splay(y,son[x][1]);
    z=son[y][0];son[y][0]=0;fa[z]=sz[z]=0;
    pushup(y);pushup(x);
}

void Move(int k,int w)
{
    int x,y,z=pos[k],rank;
    Splay(z,rt);rank=sz[son[z][0]]+1;
    Del(rank);
    if(w==INF) x=Find(rt,n),y=Find(rt,n+1);
    else if(w==-INF) x=Find(rt,1),y=Find(rt,2);
    else x=Find(rt,rank+w-1),y=Find(rt,rank+w);
    Splay(x,rt);Splay(y,son[x][1]);
    sz[z]=1;fa[z]=y;son[y][0]=z;
    pushup(y);pushup(x);
}

int main()
{
    n=read();m=read();
    for(int i=2;i<=n+1;i++) pos[num[i]=read()]=i;
    Build(1,n+2,0);rt=n+3>>1;
    for(int i=1;i<=m;i++)
    {
        char ch[10];int x;
        scanf("%s",ch);x=read();
        if(ch[0]=='Q') printf("%d\n",num[Find(rt,x+1)]);
        else if(ch[0]=='A') Ask(pos[x]);
        else if(ch[0]=='T') Move(x,-INF);
        else if(ch[0]=='B') Move(x,INF);
        else Move(x,read());
    }
    return 0;
}