1. 程式人生 > >ZJOI 2006 書架 Splay

ZJOI 2006 書架 Splay

題目連結點我點我:-)
題目描述
小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;
}