1. 程式人生 > >BZOJ1861[Zjoi2006]書架——非旋轉treap

BZOJ1861[Zjoi2006]書架——非旋轉treap

提示 i++ 裏的 hup sin 看書 可能 吸引力 pri

題目描述

小T有一個很大的書櫃。這個書櫃的構造有些獨特,即書櫃裏的書是從上至下堆放成一列。她用1到n的正整數給每本書都編了號。 小T在看書的時候,每次取出一本書,看完後放回書櫃然後再拿下一本。由於這些書太有吸引力了,所以她看完後常常會忘記原來是放在書櫃的什麽位置。不過小T的記憶力是非常好的,所以每次放書的時候至少能夠將那本書放在拿出來時的位置附近,比如說她拿的時候這本書上面有X本書,那麽放回去時這本書上面就只可能有X-1、X或X+1本書。 當然也有特殊情況,比如在看書的時候突然電話響了或者有朋友來訪。這時候粗心的小T會隨手把書放在書櫃裏所有書的最上面或者最下面,然後轉身離開。 久而久之,小T的書櫃裏的書的順序就會越來越亂,找到特定的編號的書就變得越來越困難。於是她想請你幫她編寫一個圖書管理程序,處理她看書時的一些操作,以及回答她的兩個提問:(1)編號為X的書在書櫃的什麽位置;(2)從上到下第i本書的編號是多少。

輸入

第一行有兩個數n,m,分別表示書的個數以及命令的條數;第二行為n個正整數:第i個數表示初始時從上至下第i個位置放置的書的編號;第三行到m+2行,每行一條命令。命令有5種形式: 1. Top S——表示把編號為S的書房在最上面。 2. Bottom S——表示把編號為S的書房在最下面。 3. Insert S T——T∈{-1,0,1},若編號為S的書上面有X本書,則這條命令表示把這本書放回去後它的上面有X+T本書; 4. Ask S——詢問編號為S的書的上面目前有多少本書。 5. Query S——詢問從上面數起的第S本書的編號。

輸出

對於每一條Ask或Query語句你應該輸出一行,一個數,代表詢問的答案。

樣例輸入

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

樣例輸出

2
9
9
7
5
3

提示

數據範圍

100%的數據,n,m < = 80000 因為每個編號唯一,用map存一下每個編號的書對應的平衡樹中節點編號。 這幾個操作如果知道操作點在平衡樹中的具體位置就很容易實現了,直接分裂平衡樹之後按要求做即可。
但我們只知道操作點在平衡樹中的編號而不知道在平衡樹中具體位置,所以用f[]數組存一下平衡樹上每個點的父親然後反向分裂即可(詳細實現參見平衡樹講解)。
#include<set>
#include<map>
#include<queue>
#include<stack>
#include<cmath>
#include<cstdio>
#include<vector>
#include<bitset>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
map<int,int>s;
int n,m;
int x,y;
char ch[10];
int ls[100010];
int rs[100010];
int size[100010];
int f[100010];
int r[100010];
int v[100010];
int cnt;
int root;
int a,b,c,d;
int build(int val)
{
    int rt=++cnt;
    size[rt]=1;
    v[rt]=val;
    r[rt]=rand();
    return rt;
}
void pushup(int rt)
{
    size[rt]=size[ls[rt]]+size[rs[rt]]+1;
}
int merge(int x,int y)
{
    if(!x||!y)
    {
        return x+y;
    }
    if(r[x]<r[y])
    {
        rs[x]=merge(rs[x],y);
        f[rs[x]]=x;
        pushup(x);
        if(!rs[x])
        {
            f[rs[x]]=0;
        }
        return x;
    }
    else
    {
        ls[y]=merge(x,ls[y]);
        f[ls[y]]=y;
        pushup(y);
        if(!ls[y])
        {
            f[ls[y]]=0;
        }
        return y;
    }
}
void split1(int rt,int &a,int &b)
{
    int x=ls[rt];
    int y=rs[rt];
    ls[rt]=rs[rt]=0;
    size[rt]=1;
    while(f[rt])
    {
        if(rt==ls[f[rt]])
        {
            ls[f[rt]]=y;
            f[y]=f[rt];
            y=f[rt];
            pushup(f[rt]);
        }
        else
        {
            rs[f[rt]]=x;
            f[x]=f[rt];
            x=f[rt];
            pushup(f[rt]);
        }
        rt=f[rt];
    }
    a=x;
    b=y;
}
void split2(int rt,int &x,int &y,int k)
{
    if(!rt)
    {
        x=y=0;
        return ;
    }
    if(size[ls[rt]]>=k)
    {
        y=rt;
        split2(ls[rt],x,ls[y],k);
        f[ls[y]]=y;
        pushup(rt);
    }
    else
    {
        x=rt;
        split2(rs[rt],rs[x],y,k-size[ls[rt]]-1);
        f[rs[x]]=x;
        pushup(rt);
    }
}
int query(int rt,int k)
{
    if(size[ls[rt]]>=k)
    {
        return query(ls[rt],k);
    }
    else if(size[ls[rt]]+1==k)
    {
        return rt;
    }
    else
    {
        return query(rs[rt],k-size[ls[rt]]-1);
    }
}
int main()
{
    srand(12378);
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&x);
        s[x]=build(x);
        root=merge(root,s[x]);
    }
    while(m--)
    {
        scanf("%s",ch);
        if(ch[0]==‘I‘)
        {
            scanf("%d%d",&x,&y);
            if(y==0)
            {
                continue;
            }
            x=s[x];
            split1(x,a,b);
            int num=size[a];
            root=merge(merge(a,x),b);
            if(y==-1)
            {
                split2(root,a,b,num-1);
                split2(b,b,c,1);
                split2(c,c,d,1);
                root=merge(merge(a,c),merge(b,d));
            }
            else
            {
                split2(root,a,b,num);
                split2(b,b,c,1);
                split2(c,c,d,1);
                root=merge(merge(a,c),merge(b,d));
            }
        }
        else if(ch[0]==‘T‘)
        {
            scanf("%d",&x);
            x=s[x];
            split1(x,a,b);
            root=merge(x,merge(a,b));
        }
        else if(ch[0]==‘B‘)
        {
            scanf("%d",&x);
            x=s[x];
            split1(x,a,b);
            root=merge(merge(a,b),x);
        }
        else if(ch[0]==‘A‘)
        {
            scanf("%d",&x);
            x=s[x];
            split1(x,a,b);
            printf("%d\n",size[a]);
            root=merge(merge(a,x),b);
        }
        else
        {   
            scanf("%d",&x);
            printf("%d\n",v[query(root,x)]);
        }
    }
}

BZOJ1861[Zjoi2006]書架——非旋轉treap