hdu 3487 Play with Chain(伸展树)

此题乃传说中splay的简单应用。

不愧是传说中的数据结构，花了两天才写出来，第一次写debug可是费了不少功夫啊。

老实说写这种数据结构还真是不简单，上次学treap的时候也是花了好大功夫。  

不过最终好歹是写出来了，而且学到了不少东西。算是值了。

 

上次写treap的时候，旋转操作写了两个函数，这次学会了一种新写法，可以将旋转操作直接写成一个函数。

 

在结构体中  用ch[2]  来记录儿子节点。

用ch[0]表示左儿子，用ch[1]表示右儿子。

 

于是rotate操作就可以写成

 

 

void rotate(int x, int s) { int p = tree[x].pra; tree[p].ch[s] = tree[x].ch[!s]; if (tree[x].ch[!s]) tree[tree[x].ch[!s]].pra = p; tree[x].ch[!s] = p; tree[x].pra = tree[p].pra; if (tree[x].pra != -1) tree[tree[x].pra].ch[tree[tree[x].pra].ch[1] == p] = x; tree[p].pra = x; tree[x].size = tree[p].size; update(p); } 

其中s=0 表示x节点是其父节点的左儿子    s=1则表示x是其父节点的右儿子

 

这样一来两种旋转操作就可以用一个函数搞定了。     

 

至于伸展树的核心操作splay(int x,int f)   将x节点旋转到f节点的下方。则是两种原始旋转操作的组合，注意一字型旋转和之字型旋转时，旋转节点的顺序就行了。

 

 

有一种写法很不容易想到，就是标记法。

比如  将某一区间的数字反转，我们再提取该区间后，不用马上将该区间所有数进行反转操作，只需用一个标记标记该区间在访问时候需要反转。   这样的话，可以省很多时间。因为不访问时，进行反转操作是没有必要的。   比如在该区间被访问之前需要经历2×n次反转，那么实际上该区间最后的效果是不需要反转，我们只要修改标记就可以达到这个效果;

当访问到该区间时，我们对该节点执行标记的相应操作，然后将标记下移即可。    

 

如果写过线段树的话，这点应该很容易理解了。

 

如果想学习splay ，可以看看杨思雨的论文，baidu一下就出来了

 

 

贴下我自己的代码吧：

跑了600多MS，写的挺一般的。    

/* * File: main.cpp * Author: liuwei * * Created on September 22, 2010, 2:30 PM */ #include <cstdlib> #include <cstdio> #include <iostream> using namespace std; const int MAXN = 300000; int num[MAXN + 10]; void inOrder(int x); struct node { int val; int size; int ch[2]; int pra; bool rev; void init() { ch[0] = ch[1] = 0; size = 1; rev = 0; } } tree[MAXN + 10]; int len; int root; int n, m; inline void swap(int &a, int &b) { int tmp = a; a = b; b = tmp; return; } inline int getSize(int x) { if (x) return tree[x].size; return 0; } inline void update(int x) { tree[x].size = 1 + getSize(tree[x].ch[0]) + getSize(tree[x].ch[1]); } void build(int &x, int s, int e, int f) { if (s > e) return; int mid = (s + e) >> 1; x = ++len; tree[x].val = num[mid]; tree[x].init(); tree[x].pra = f; build(tree[x].ch[0], s, mid - 1, x); build(tree[x].ch[1], mid + 1, e, x); update(x); return; } void init(int n) { tree[0].pra = -1; len = 0; int i(1); for (; i <= n; ++i) { num[i] = i; } num[0] = num[n + 1] = -1; build(tree[0].ch[1], 0, n + 1, 0); //insert n+2 number } void rotate(int x, int s) { int p = tree[x].pra; tree[p].ch[s] = tree[x].ch[!s]; if (tree[x].ch[!s]) tree[tree[x].ch[!s]].pra = p; tree[x].ch[!s] = p; tree[x].pra = tree[p].pra; if (tree[x].pra != -1) tree[tree[x].pra].ch[tree[tree[x].pra].ch[1] == p] = x; tree[p].pra = x; tree[x].size = tree[p].size; update(p); } void splay(int x, int f) { while (tree[x].pra != f) { int p = tree[x].pra; if (tree[p].pra == f) { if (tree[p].ch[0] == x) rotate(x, 0); else rotate(x, 1); } else { int g = tree[p].pra; if (tree[g].ch[0] == p) { if (tree[p].ch[0] == x) rotate(p, 0), rotate(x, 0); else rotate(x, 1), rotate(x, 0); } else { if (tree[p].ch[1] == x) rotate(p, 1), rotate(x, 1); else rotate(x, 0), rotate(x, 1); } } } return; } inline void pushDown(int x) { if (x && tree[x].rev) { swap(tree[x].ch[0], tree[x].ch[1]); tree[x].rev ^= 1; tree[tree[x].ch[0]].rev ^= 1; tree[tree[x].ch[1]].rev ^= 1; } } void select(int k, int f) { int x = tree[0].ch[1]; pushDown(x); while (k != getSize(tree[x].ch[0]) + 1) { if (k <= getSize(tree[x].ch[0])) x = tree[x].ch[0]; else { k -= getSize(tree[x].ch[0]) + 1; x = tree[x].ch[1]; } pushDown(x); } splay(x, f); return; } void reverse() { int a, b; scanf("%d%d", &a, &b); select(a, 0); root = tree[0].ch[1]; select(b + 2, root); int tmp = tree[tree[root].ch[1]].ch[0]; tree[tmp].rev ^= 1; return; } int cou; void inOrder(int x) { pushDown(x); if (tree[x].ch[0]) inOrder(tree[x].ch[0]); if (cou >= 1 && cou <= n) { if (cou > 1) printf(" "); printf("%d", tree[x].val); } ++cou; if (tree[x].ch[1]) inOrder(tree[x].ch[1]); } void cut() { int a, b, c; scanf("%d%d%d", &a, &b, &c); select(a, 0); root = tree[0].ch[1]; select(b + 2, root); int tmp = tree[tree[root].ch[1]].ch[0]; tree[tree[root].ch[1]].ch[0] = 0; update(tree[root].ch[1]); update(root); select(c + 1, 0); root = tree[0].ch[1]; select(c + 2, root); tree[tree[root].ch[1]].ch[0] = tmp; tree[tmp].pra = tree[root].ch[1]; update(tree[root].ch[1]); update(root); return; } int main(int argc, char** argv) { while (EOF != scanf("%d%d", &n, &m)) { if (n == -1 && m == -1) break; init(n); char cmd[10]; while (m--) { scanf("%s", cmd); if (cmd[0] == 'C') cut(); else reverse(); } cou = 0; inOrder(tree[0].ch[1]); printf("/n"); } return 0; }