[bzoj1861][Splay]Book 书架

6 篇文章 0 订阅

Description

小T有一个很大的书柜。这个书柜的构造有些独特,即书柜里的书是从上至下堆放成一列。她用1到n的正整数给每本书都编了号。
小T在看书的时候,每次取出一本书,看完后放回书柜然后再拿下一本。由于这些书太有吸引力了,所以她看完后常常会忘记原来是放在书柜的什么位置。不过小T的记忆力是非常好的,所以每次放书的时候至少能够将那本书放在拿出来时的位置附近,比如说她拿的时候这本书上面有X本书,那么放回去时这本书上面就只可能有X-1、X或X+1本书。
当然也有特殊情况,比如在看书的时候突然电话响了或者有朋友来访。这时候粗心的小T会随手把书放在书柜里所有书的最上面或者最下面,然后转身离开。
久而久之,小T的书柜里的书的顺序就会越来越乱,找到特定的编号的书就变得越来越困难。于是她想请你帮她编写一个图书管理程序,处理她看书时的一些操作,以及回答她的两个提问:(1)编号为X的书在书柜的什么位置;(2)从上到下第i本书的编号是多少。

Input

第一行有两个数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本书的编号。

Output

对于每一条Ask或Query语句你应该输出一行,一个数,代表询问的答案。

Sample Input

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

Sample Output

2

9

9

7

5

3

HINT

数据范围

100%的数据,n,m < = 80000

题解

Splay维护这个序列
记录每个编号的书在Splay中对应的编号
前三个操作的话直接把节点删掉然后暴力插入
后两个就基本操作2333..

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
struct node
{
    int f,d,c,son[2];
}tr[210000];int len,root;
void add(int d,int f,int pla)
{
    len++;tr[len].d=d;tr[len].f=f;tr[len].c=1;
    tr[len].son[0]=tr[len].son[1]=0;
    tr[f].son[pla]=len;
}
void upd(int now){tr[now].c=tr[tr[now].son[0]].c+tr[tr[now].son[1]].c+1;}
void rotate(int x,int w)
{
    int f=tr[x].f,ff=tr[f].f;
    int R,r;
    R=f;r=tr[x].son[w];
    tr[R].son[1-w]=r;
    if(r!=0)tr[r].f=R;
    R=ff;r=x;
    if(tr[R].son[0]==f)tr[R].son[0]=r;
    else tr[R].son[1]=r;
    tr[r].f=R;
    R=x;r=f;
    tr[R].son[w]=r;
    tr[r].f=R;
    upd(f);upd(x);
}
void splay(int x,int rt)
{
    while(tr[x].f!=rt)
    {
        int f=tr[x].f,ff=tr[f].f;
        if(ff==rt)
        {
            if(tr[f].son[0]==x)rotate(x,1);
            else rotate(x,0);
        }
        else
        {
            if(tr[ff].son[0]==f && tr[f].son[0]==x)rotate(f,1),rotate(x,1);
            else if(tr[ff].son[1]==f && tr[f].son[0]==x)rotate(x,1),rotate(x,0);
            else if(tr[ff].son[1]==f && tr[f].son[1]==x)rotate(f,0),rotate(x,0);
            else rotate(x,0),rotate(x,1);
        }
    }
    if(rt==0)root=x;
}
void del(int x)
{
    splay(x,0);
    if(tr[x].son[0]==0 && tr[x].son[1]==0){root=len=0;return ;}
    else if(tr[x].son[0]!=0 && tr[x].son[1]==0){root=tr[x].son[0];tr[root].f=0;}
    else if(tr[x].son[0]==0 && tr[x].son[1]!=0){root=tr[x].son[1];tr[root].f=0;}
    else
    {
        int p=tr[x].son[0];
        while(tr[p].son[1]!=0)p=tr[p].son[1];
        splay(p,x);
        int R=p,r=tr[x].son[1];
        tr[R].son[1]=r;tr[r].f=R;
        tr[p].f=0;root=p;
        upd(p);
    }
}
int findKth(int pos)
{
    int x=root;
    while(1)
    {
        int lc=tr[x].son[0],rc=tr[x].son[1];
        if(pos<=tr[lc].c)x=lc;
        else if(pos>tr[lc].c+1)pos-=tr[lc].c+1,x=rc;
        else return x;
    }
}
int ins(int op,int pos,int tmp)//把一个权为op的点 插入到pos位置  成为他的tmp孩子 
{
    int x=findKth(pos);
    if(tr[x].son[tmp]==0)add(op,x,tmp);
    else
    {
        int r=tr[x].son[tmp];
        add(op,x,tmp);upd(len);
        tr[r].f=len;
        tr[len].son[tmp]=r;
    }
    splay(len,0);
    return len;
}
int bel[110000],col[110000];
int maketree(int l,int r)
{
    if(l>r)return 0;
    if(l==r)
    {
        len++;bel[col[l]]=len;
        tr[len].d=col[l];tr[len].c=1;
        tr[len].son[0]=tr[len].son[1]=0;return len;
    }
    int mid=(l+r)/2;
    int lc=maketree(l,mid-1);
    int now=++len;
    bel[col[mid]]=now;
    tr[now].d=col[mid];tr[now].c=1;
    tr[now].son[0]=lc;
    int rc=maketree(mid+1,r);
    tr[now].son[1]=rc;
    if(lc)tr[lc].f=now;
    if(rc)tr[rc].f=now;
    upd(now);
    return now;
}
int n,m;
char ch[15];
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)scanf("%d",&col[i]);
    root=maketree(1,n);
    while(m--)
    {
        int u,c;scanf("%s%d",ch+1,&u);
        if(ch[1]=='T')
        {
            int tmp=bel[u];del(tmp);
            bel[u]=ins(u,1,0);
        }
        else if(ch[1]=='B')
        {
            int tmp=bel[u];del(tmp);
            bel[u]=ins(u,n-1,1);
        }
        else if(ch[1]=='I')
        {
            scanf("%d",&c);
            int x=bel[u];splay(x,0);
            int pos=tr[tr[x].son[0]].c;
            del(x);
            pos+=c;
            if(pos>=1 && pos<=(n-1))bel[u]=ins(u,pos,1);
            if(pos==0)bel[u]=ins(u,1,0);
        }
        else if(ch[1]=='A')
        {
            int x=bel[u];
            splay(x,0);
            printf("%d\n",tr[tr[x].son[0]].c);
        }
        else printf("%d\n",tr[findKth(u)].d);
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值