hdu4453 伸展树基本题

题意：很多操作，具体题目中有图http://acm.hdu.edu.cn/showproblem.php?pid=4453

move光标操作，move1， move2， 我们假定伸展树的第一个点为光标的位置，那么假如光标向后移动，我们可以把第一个数删除然后插入到整个序列的最后，同理光标向前移动也差不多，其它操作都是很常见的操作。

伸展树敲的不多，花了一点时间调试，终于AC了。

现在正式比赛估计这种题都会被顺秒了吧，高手越来越多了。

#include <cstdio>#include <cstring>#include <algorithm>using namespace std;#define KT ch[ ch[root][1] ][0]#define L ch[x][0]#define R ch[x][1]const int maxn = 300005;int num[maxn], n, m;int k1, k2;typedef long long ll;struct splayTree {    int ch[maxn][2], sz[maxn], pre[maxn];    int root, tot, all; // all：节点总数， tot：最大标号    int add[maxn], val[maxn];    bool flip[maxn]; //翻转标记    int sta[maxn], top;    void rotate(int &x, int f) {           int y = pre[x], z = pre[y];           down(y); down(x);           ch[y][!f] = ch[x][f];           pre[ch[x][f]] = y;           pre[x] = pre[y];           if(z) ch[z][ch[z][1] == y] = x;           ch[x][f] = y;           pre[y] = x;           up(y);       }       void splay(int &x, int g) {           down(x);           while(pre[x] != g) {               int y = pre[x], z = pre[y];               down(z); down(y); down(x);               if(z == g) {                   rotate(x, ch[y][0] == x);               }               else {                   int f = (ch[z][0] == y);                   ch[y][!f] == x ? rotate(y, f) : rotate(x, !f);                   rotate(x, f);               }           }           if(!g) root = x;           up(x);       }       void rto(int k, int g) {           int x = root;           while(1) {               down(x);               if(sz[L] == k) break;               if(sz[L] > k) x = L;               else {                   k -= sz[L] + 1;                   x = R;               }           }           splay(x, g);       }    void down(int x){        if(add[x]) {            val[L] += add[x];            val[R] += add[x];            add[L] += add[x];            add[R] += add[x];            add[x] = 0;        }        if(flip[x]) {            flip[L] ^= 1;            flip[R] ^= 1;            swap(L, R);            flip[x] = 0;        }    }    void up(int x) {        sz[x] = sz[L] + sz[R] +1;    }    void build(int &x, int l, int r, int fa) {        if(l > r) return;        int m = (l+r) >> 1;        newNode(x, num[m], fa);        build(L, l, m-1, x);        build(R, m+1, r, x);        up(x);    }    void newNode(int &x, int v, int fa) {        if(top) x = sta[top--];//内存回收        else x = ++tot;        all++;        pre[x] = fa;        sz[x] = 1;        L = R = 0;        val[x] = v;        add[x] = 0;        flip[x] = 0;    }    void init(int n) {        all = tot = top = 0;        newNode(root, 0, 0);        newNode(ch[root][1], 0, root);        for(int i = 0; i < n; i++)            scanf("%d", &num[i]);        build(KT, 0, n-1, ch[root][1]);        up(ch[root][1]);        up(root);    }    ll erase(int k) { //删除第k个数        rto(k, 0);        rto(k-1, root);        sta[++top] = root;        all--;        ll ret = val[root];        int ls = ch[root][0], rs = ch[root][1];        root = ls;        pre[ls] = 0;        ch[ls][1] = rs;        if(rs)pre[rs] = ls;        up(root);        return ret;    }    void insert(int k, int v) { //在第k个数后面插入一个数v        rto(k, 0);        int x;        int rs = ch[root][1];        newNode(x, v, root);        ch[root][1] = x;        ch[x][1] = rs;        if(rs) pre[ch[x][1]] = x;        up(ch[root][1]);        up(root);    }    void move1() {        int v = erase(all-2);        insert(0, v);    }    void move2() {        int v = erase(1);        insert(all-2, v);    }    void update(int l, int r, int v) {        rto(l-1, 0);        rto(r+1, root);        val[KT] += v;        add[KT] += v;        up(ch[root][1]);        up(root);    }    void reverse(int l, int r) {        rto(l-1, 0);        rto(r+1, root);        flip[KT] ^= 1;        up(ch[root][1]);        up(root);    }    int query() {        rto(1, 0);        return val[root];    }    void print(int x) {        printf("node %d: ls=%d rs=%d lsz = %d rsz = %d val = %d\n", x, L, R, sz[L], sz[R], val[x]);        if(L)print(L);        if(R)print(R);    }    void debug() {        printf("root = %d\n", root);         print(root);    }}spt;int main() {    int ca = 1;    while(~scanf("%d%d%d%d", &n, &m, &k1, &k2)) {        if(!m && !n && !k1 && !k2) break;        char op[111];        spt.init(n);      //  spt.debug();        printf("Case #%d:\n", ca++);        while(m--) {            scanf("%s", op);            if(op[0] == 'q') {                printf("%d\n", spt.query());            }            if(op[0] == 'm') {                int kd;                scanf("%d", &kd);                if(kd == 1) spt.move1();                else spt.move2();            }            if(op[0] == 'i') {                int v;                scanf("%d", &v);                spt.insert(1, v);            }            if(op[0] == 'd')                spt.erase(1);            if(op[0] == 'a') {                int v;                scanf("%d", &v);                spt.update(1, k2, v);            }            if(op[0] == 'r')                spt.reverse(1, k1);        }    }    return 0;}/*5 100000 2 41 2 3 4 5*/