题解:平衡树模板题 照题意模拟即可
/**************************************************************
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
",x,ch[x][0],ch[x][1],pre[x],size[x],fp[x]);
Treavel(ch[x][1]);
}
}
void debug(int rp)
{
printf("root:%d
",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
",operator4(s));
else s=read(),printf("%d
",operator5(s));
}
return 0;
}
1861: [Zjoi2006]Book 书架
Time Limit: 4 Sec Memory Limit: 64 MBSubmit: 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
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
9
9
7
5
3
HINT
数据范围
100%的数据，n,m < = 80000