BZOJ1861: [Zjoi2006]Book 书架

题解:平衡树模板题 照题意模拟即可

/**************************************************************
    Problem: 1861
    User: c20161007
    Language: C++
    Result: Accepted
    Time:1516 ms
    Memory:4256 kb
****************************************************************/
 
#include <bits/stdc++.h>
const int MAXN=8e4+10;
#define ll long long
using namespace std;
ll read(){
    ll x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
    while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
    return f*x;
}
int pre[MAXN],size[MAXN],ch[MAXN][2],p[MAXN],fp[MAXN],cnt,a[MAXN];
int rt,n,q;
vector<int>vec;
void Treavel(int x)
{
    if(x)
    {
        Treavel(ch[x][0]);
        printf("结点%2d:左儿子 %2d 右儿子 %2d 父结点 %2d size=%2d key=%2d\n",x,ch[x][0],ch[x][1],pre[x],size[x],fp[x]);
        Treavel(ch[x][1]);
    }
}
void debug(int rp)
{
    printf("root:%d\n",rp);
    Treavel(rp);
}
void newnode(int &x,int y){
    x=++cnt;size[x]=1;ch[x][0]=ch[x][1]=0;pre[x]=y;
}
void up(int x){size[x]=size[ch[x][0]]+size[ch[x][1]]+1;}
void built(int &x,int l,int r,int fa){
    if(l>r)return ;
    int mid=(l+r)>>1;
    newnode(x,fa);p[a[mid]]=cnt;fp[cnt]=a[mid];
    built(ch[x][0],l,mid-1,x);
    built(ch[x][1],mid+1,r,x);
    up(x);
}
void inte(){
    newnode(rt,0);
    newnode(ch[rt][1],rt);
    built(ch[ch[rt][1]][0],1,n,ch[rt][1]);
    up(ch[rt][1]);up(rt);
  //  debug(rt);
}
void rotate(int x,int kind){
    int y=pre[x];
    pre[ch[x][kind]]=y;ch[y][!kind]=ch[x][kind];
    if(pre[y])ch[pre[y]][ch[pre[y]][1]==y]=x;
    pre[x]=pre[y];ch[x][kind]=y;pre[y]=x;
    up(y);
}
void splay(int x,int goal){
    while(pre[x]!=goal){
    if(pre[pre[x]]==goal)rotate(x,ch[pre[x]][0]==x);
    else{
        int y=pre[x];int kind=ch[pre[y]][0]==y;
        if(ch[y][kind]==x)rotate(x,!kind),rotate(x,kind);
        else rotate(y,kind),rotate(x,kind);
    }
    }
    if(goal==0)rt=x;
    up(x);
}
int find1(int x,int sz){
    if(size[ch[x][0]]+1==sz)return x;
    else if(size[ch[x][0]]>=sz)return find1(ch[x][0],sz);
    else return find1(ch[x][1],sz-size[ch[x][0]]-1);
}
void destry(int x){
    splay(x,0);int sz=size[ch[x][0]];
    splay(find1(rt,sz),0);splay(find1(rt,sz+2),rt);
    ch[ch[rt][1]][0]=0;up(ch[rt][1]);up(rt);
}
void operator1(int s){
    int pos=p[s];destry(pos);
    splay(find1(rt,1),0);splay(find1(rt,2),rt);
    ch[ch[rt][1]][0]=pos;ch[pos][0]=ch[pos][1]=0;size[pos]=1;pre[pos]=ch[rt][1];
    up(ch[rt][1]);up(rt);
}
void operator2(int s){
    int pos=p[s];destry(pos);
    splay(find1(rt,size[rt]-1),0);splay(find1(rt,size[rt]),rt);
    ch[ch[rt][1]][0]=pos;ch[pos][0]=ch[pos][1]=0;size[pos]=1;pre[pos]=ch[rt][1];
    up(ch[rt][1]);up(rt);
}
void operator3(int s,int t){
    int pos=p[s];splay(pos,0);
 //   cout<<pos<<endl;
    int sz=size[ch[rt][0]]-1;
   // debug(rt);
    if(sz==0&&t==-1||sz==size[rt]-3&&t==1||t==0)return ;
    destry(pos);
    sz++;sz+=t;
   // cout<<sz<<" "<<sz<<endl;
    //cout<<find1(rt,sz)<<" "<<vec[fp[find1(rt,sz+1)]-1]<<endl;
   // debug(rt);
    splay(find1(rt,sz),0);splay(find1(rt,sz+1),rt);
    ch[ch[rt][1]][0]=pos;ch[pos][0]=ch[pos][1]=0;size[pos]=1;pre[pos]=ch[rt][1];
    up(ch[rt][1]);up(rt);
}
int operator4(int s){
    splay(p[s],0);
    return size[ch[rt][0]]-1;
}
int operator5(int s){
    return vec[fp[find1(rt,s+1)]-1];
}
char str[11];
int get_id(int x,int t){return lower_bound(vec.begin(),vec.begin()+t,x)-vec.begin()+1;}
int main(){
    n=read();q=read();
    for(int i=1;i<=n;i++)a[i]=read(),vec.push_back(a[i]);
    sort(vec.begin(),vec.end());
    int sz=unique(vec.begin(),vec.end())-vec.begin();
    for(int i=1;i<=n;i++)a[i]=lower_bound(vec.begin(),vec.begin()+sz,a[i])-vec.begin()+1;
    inte();int s,t;
    for(int i=1;i<=q;i++){
    scanf(" %s",str);
    if(str[0]=='T')s=read(),operator1(s);
    else if(str[0]=='B')s=read(),s=get_id(s,sz),operator2(s);
    else if(str[0]=='I')s=read(),s=get_id(s,sz),t=read(),operator3(s,t);
    else if(str[0]=='A')s=read(),s=get_id(s,sz),printf("%d\n",operator4(s));
    else s=read(),printf("%d\n",operator5(s));
    }
    return 0;
}

 

1861: [Zjoi2006]Book 书架

Time Limit: 4 Sec  Memory Limit: 64 MB
Submit: 2249  Solved: 1312
[Submit][Status][Discuss]

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