ZJOI 2006 书架 Splay

来源:互联网 发布:大数据架构基础 编辑:程序博客网 时间:2024/06/06 20:29

题目链接:点我点我:-)
题目描述
小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语句你应该输出一行,一个数,代表询问的答案。

思路
splay模板题
1.对于top操作,将s旋转至根,如果s没有左子树,则返回。如有,并且其无右子树,则将左子树放到右子树。如左右子树均有,则找到右子树的最左下方的点t,并把左子树放到t的左下方。(bottom操作反之)
2.对于insert操作,就是与其前驱或后驱交换位置。找前驱的话,将s旋转至根,去找左子树最右下方的点(后驱则反之)
3.Ask操作我们只需要维护一下编号,所以再开一个数组维护下
4.Query操作为splay基本操作

感想
有没有真的可以搞清楚splay的写法用法?(严重注意本题看了标程还调了好久)啊加上已经做了的网络的splay做法应该差不多的吧,可以再实战一下

错误整理
很多很多:
1.把top操作的左子树放到右子树写成了右子树放到左子树
2.然后,放完了以后,原子树忘记清空了。。。
3.最后,swap中数组的下标写反了。。
。。。没见过我这么聪明的人。。居然4个小时就调出来了。。。

代码

//miaomiao 2017.2.3#include<cstdio>#include<cstdlib>#include<cstring>#include<algorithm>using namespace std;#define lc ch[o][0]#define rc ch[o][1]#define For(i, a, b) for(int i = (a); i <= (int)(b); i++)#define N (80000+5)int rank[N];void swp(int &a, int &b){a^=b^=a^=b;}struct Splay{    int ch[N][2], fa[N], siz[N], val[N], rt, p;    void init(){rt = p = 0;}    void maintain(int o){        siz[o] = 1;        if(lc) siz[o] += siz[lc]; if(rc) siz[o] += siz[rc];    }    void rotate(int o){        int f = fa[o], g = fa[f];        int c = (ch[f][1]==o);        fa[f] = o; fa[o] = g;        fa[ch[o][c^1]] = f;        if(g) ch[g][ch[g][1]==f] = o;        ch[f][c] = ch[o][c^1]; ch[o][c^1] = f;        maintain(f);    }    void splay(int o, int aim){        while(fa[o] != aim){            int f = fa[o], g = fa[f];            if(g != aim) rotate(f);            rotate(o);        }        maintain(o);        if(!aim) rt = o;    }    void insert(int x){        int mf = rt;        while(ch[mf][1]) mf = ch[mf][1];        ch[mf][1] = ++p; siz[p] = 1; fa[p] = mf; rank[x] = p; val[p] = x;        splay(p, 0);     }    void Top_Bottom(int num, int tb){        int x = rank[num];        splay(x, 0);            if(!ch[x][tb]) return;        if(!ch[x][tb^1]){ch[x][tb^1] = ch[x][tb]; ch[x][tb] = 0; return;}         int ins = ch[x][tb], pos = ch[x][tb^1];        while(ch[pos][tb]) pos = ch[pos][tb];        ch[pos][tb] = ins; fa[ins] = pos; ch[x][tb] = 0;        splay(ins, 0);    }    void suf_pre(int num, int sp){        if(!sp) return;        int x = rank[num];        if(sp==-1) sp = 0;        splay(x, 0);        int pos = ch[x][sp];        while(ch[pos][sp^1]) pos = ch[pos][sp^1];        swp(rank[num], rank[val[pos]]); swp(val[x], val[pos]);    }    int Ask(int num){        int x = rank[num];        splay(x, 0);        return siz[ch[x][0]];    }    int query(int aim){        int x = rt;        while(aim){            if(siz[ch[x][0]]+1==aim) return val[x];                if(aim <= siz[ch[x][0]]) x = ch[x][0];            else aim -= siz[ch[x][0]]+1, x = ch[x][1];        }    }}SP;char op[10];int main(){#ifndef ONLINE_JUDGE    freopen("test.in", "r", stdin);    freopen("test.out", "w", stdout);#endif    SP.init();    int n, m, x, t;    scanf("%d%d", &n, &m);    For(i, 1, n){        scanf("%d", &x);        SP.insert(x);    }    For(i, 1, m){        scanf("%s%d", op, &x);        if(op[0]=='I') scanf("%d", &t);        switch(op[0]){            case 'T': SP.Top_Bottom(x, 0); break;            case 'B': SP.Top_Bottom(x, 1); break;            case 'I': SP.suf_pre(x, t); break;            case 'A': printf("%d\n", SP.Ask(x)); break;            case 'Q': printf("%d\n", SP.query(x)); break;        }    }    return 0;}
1 0
原创粉丝点击