码迷,mamicode.com
首页 > 其他好文 > 详细

[ZJOI2006]书架

时间:2018-03-13 01:01:50      阅读:210      评论:0      收藏:0      [点我收藏+]

标签:特殊情况   吸引   scanf   uil   namespace   root   取出   can   --   

题目描述

小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语句你应该输出一行,一个数,代表询问的答案。

 

输入输出样例

输入样例#1: 
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: 
2
9
9
7
5
3

说明

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

 

 

都是splay的基本操作,然而我都调成儿子了2333

总结一下小错:

1.变量写混(常规操作)

2.把x插入到第k大后面的时候忘了更新size(也差不多是常规操作了)

3.最致命的是我删除的时候如果被删除的元素是最大或者最小的话,我把根设成了前驱(后继),而不是它们在树上的儿子2333,这个错的真心玄学。

 

所以以后一定要,think twice ,code once.

 

#include<bits/stdc++.h>
#define ll long long
#define maxn 100005
using namespace std;
int f[maxn],ch[maxn][2];
int a[maxn],siz[maxn];
int n,m,pos,root,lefbig,rigsml;
int cnt,S,T,u,p;
char s[23];

inline void update(int x){
    siz[x]=siz[ch[x][0]]+siz[ch[x][1]]+1;
}

inline int get(int x){
    return ch[f[x]][1]==x;
}

inline void rotate(int x){
    int fa=f[x],ffa=f[fa],tp=get(x);
    ch[fa][tp]=ch[x][tp^1],f[ch[fa][tp]]=fa;
    ch[x][tp^1]=fa,f[fa]=x;
    f[x]=ffa;
    if(ffa) ch[ffa][ch[ffa][1]==fa]=x;
    if(root==fa) root=x;
    
    update(fa),update(x);
}

inline void splay(int x,int d){
    for(int fa;(fa=f[x])!=d;rotate(x))
        if(f[fa]!=d) rotate(get(fa)==get(x)?fa:x);
}

inline int pre(int x){
    x=ch[x][0];
    while(ch[x][1]) x=ch[x][1];
    return x;
}

inline int nxt(int x){
    x=ch[x][1];
    while(ch[x][0]) x=ch[x][0];
    return x;
}

inline void del(int x){
    splay(x,0);
    lefbig=pre(root),rigsml=nxt(root);
    if(!lefbig||!rigsml){
        if(!lefbig&&!rigsml) root=0;
        else if(!lefbig) f[ch[root][1]]=0,root=ch[root][1];
        else f[ch[root][0]]=0,root=ch[root][0];
    }
    else{
        splay(lefbig,0),splay(rigsml,root);
        ch[ch[root][1]][0]=0,update(ch[root][1]),update(root);
    }
    
    siz[x]=f[x]=ch[x][0]=ch[x][1]=0;
}

inline int rank(int x){
    int an=siz[ch[x][0]],fa;
    while(x){
        fa=f[x];
        if(get(x)) an+=siz[ch[fa][0]]+1;
        x=fa;
    }
    return an;
}

inline int BOOK(int x){
    int now=root;
    while(now){
        if(siz[ch[now][0]]>=x) now=ch[now][0];
        else{
            x-=siz[ch[now][0]]+1;
            if(!x) return now;
            now=ch[now][1];
        }
    }
}

inline void ins(int x,int F){
    int now=root,fa=0,las;
    while(now){
    	siz[now]++;
        if(F<=siz[ch[now][0]]) fa=now,now=ch[now][0],las=0;
        else{
            F-=siz[ch[now][0]]+1;
            fa=now,now=ch[now][1];
            las=1;
        }
    }

    siz[x]=1,f[x]=fa,ch[fa][las]=x,update(fa);
    splay(x,0);
}

int build(int l,int r,int fa){
    if(l>r) return 0;
    int mid=l+r>>1,O=a[mid];
    f[O]=fa;
    ch[O][0]=build(l,mid-1,O);
    ch[O][1]=build(mid+1,r,O);
    update(O);
    
    return O;
}

void print(int x){
    if(!x) return;
    print(ch[x][0]);
    printf("%d ",x);
    print(ch[x][1]);
}

int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",a+i);
    root=build(1,n,0);
    while(m--){
        scanf("%s",s);
        if(s[0]==‘I‘){
            scanf("%d%d",&S,&T);
            u=rank(S)+T;
            del(S);
            ins(S,u);
        }
        else if(s[0]==‘T‘){
            scanf("%d",&S);
            del(S);
            ins(S,0);
        }
        else if(s[0]==‘B‘){
            scanf("%d",&S);
            del(S);
            ins(S,n-1);
        }
        else if(s[0]==‘A‘){
            scanf("%d",&S);
            printf("%d\n",rank(S));
        }
        else{
            scanf("%d",&S);
            printf("%d\n",BOOK(S));
        }
	//	print(root);
	//	puts("");
    }
    
    return 0;
}

  

[ZJOI2006]书架

标签:特殊情况   吸引   scanf   uil   namespace   root   取出   can   --   

原文地址:https://www.cnblogs.com/JYYHH/p/8552610.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!