POJ 3468.A Simple Problem with Integers 解题报告

用树状数组和线段树会比较简单，这里用这道题来学习Splay。

第一次写，代码比较丑

/*
       初始化添加一个key值足够大的结点
       保证每个需要的结点都有后继
*/
#include <iostream>
#include <cstdio>
#define ll long long
using namespace std;

const int MAXN = , INF = 0x7fffffff;

struct node {
    //需要的记录信息
    ll  key, val, sum, lazy, Size, Cnt;
    //指向儿子和父亲的指针
    node *ch[], *pre;
    node() {pre = ch[] = ch [] = ; Size = ; key = ;}
    node (ll key) : key (key) {pre = ch[] = ch[] = ; Size = , Cnt = , lazy = ;}
    void Csize() {
        Size = Cnt;
        if (ch[] != NULL) Size += ch[]->Size;
        if (ch[] != NULL) Size += ch[]->Size;
    }
    void Csum() {
        sum = val;
        if (ch[] != NULL) sum += ch[]->sum + ch[]->Size * ch[]->lazy;
        if (ch[] != NULL) sum += ch[]->sum + ch[]->Size * ch[]->lazy;
    }
} nil (), *NIL = &nil;

struct Splay {
    node *root, nod[MAXN];
    int ncnt;//计算key值不同的结点数，已去重
    Splay() {
        ncnt = ;
        root = & (nod[ncnt++] = node (INF) );
        root->pre = NIL;
        root->val = root->sum = ;
    }
    void Push_Down (node *x) {
        if (x->lazy != ) {
            if (x->ch[] != NULL) x->ch[]->lazy += x->lazy;
            if (x->ch[] != NULL) x->ch[]->lazy += x->lazy;
                     x->val+=x->lazy;
        }
        x->lazy = ;
    }

    void Update (node *x) {
        x->Csize();
        x->Csum();
    }

    void Rotate (node *x, int sta) { //单旋转操作，0左旋，1右旋
        node *p = x->pre, *g = p->pre;
        Push_Down (p), Push_Down (x);
        p->ch[!sta] = x->ch[sta];
        if (x->ch[sta] != NULL)  x->ch[sta]->pre = p;
        x->pre = g;
        if (g != NIL)
            if (g->ch[] == p)  g->ch[] = x;
            else g->ch[] = x;
        x->ch[sta] = p, p->pre = x, Update (p);
        if (p == root ) root = x;
    }

    void splay (node *x, node *y) { //Splay 操作，表示把结点x,转到根
        for (Push_Down (x) ; x->pre != y;) { //将x的标记往下传
            if (x->pre->pre == y) { //目标结点为父结点
                if (x->pre->ch[] == x)       Rotate (x, );
                else   Rotate (x, );
            }
            else {
                node *p = x->pre, *g = p->pre;
                if (g->ch[] == p)
                    if (p->ch[] == x)
                        Rotate (p, ), Rotate (x, );//   / 一字型双旋转
                    else
                        Rotate (x, ), Rotate (x, );//  < 之字形双旋转

                else if (p ->ch[] == x)
                    Rotate (p, ), Rotate (x, );//    \ 一字型旋转
                else
                    Rotate (x, ), Rotate (x, ); //   >之字形旋转
            }
        }
        Update (x); //维护x结点
    }
    //找到中序便利的第K个结点，并旋转至结点y的下面。
    void Select (int k, node *y) {
        int tem ;
        node *t ;
        for ( t = root; ; ) {
            Push_Down (t) ; //标记下传
            tem = t->ch[]->Size ;
            if (k == tem +  ) break ; //找到了第k个结点 t
            if (k <= tem)
                t = t->ch[] ; //第k个结点在左子树
            else
                k -= tem +  , t = t->ch[] ;//在右子树
        }
        splay (t, y);
    }
    bool Search (ll key, node *y) {
        node *t = root;
        for (; t != NULL;) {
            Push_Down (t);
            if (t->key > key && t->ch[] != NULL) t = t->ch[];
            else if (t->key < key && t->ch[] != NULL) t = t->ch[];
            else
                break;
        }
        splay (t, y);
        return t->key == key;
    }
    void Insert (int key, int val) {
        if (Search (key, NIL) ) root->Cnt++, root->Size++;
        else {
            int d = key > root->key;
            node *t = & (nod[++ncnt] = node (key) );
            Push_Down (root);
            t->val = t->sum = val;
            t->ch[d] = root->ch[d];
            if (root->ch[d] != NULL) root->ch[d]->pre = t;
            t->ch[!d] = root;
            t->pre = root->pre;
            root->pre = t;
            root->ch[d] = NULL;
            Update (root);
            root = t;
        }
        Update (root);
    }
} sp;

ll  n, m, x;
int main() {
    scanf ("%lld %lld", &n, &m);
    sp.Insert (, );
    sp.Insert (n + , );
    for (int i = ; i <= n; i++) {
        scanf ("%lld", &x);
        sp.Insert (i, x);
    }
    char cmd;
    int l, r;
    for (int i = ; i <= m; i++) {
        scanf ("\n%c %d %d", &cmd, &l, &r);
        sp.Search (l - , NIL);
        sp.Search (r + , sp.root);
        if (cmd == 'Q') {
            node *t = sp.root->ch[]->ch[];
            ll ans = t->sum + t->Size * t->lazy;
            printf ("%lld\n", ans);
        }
        if (cmd == 'C') {
            ll c;
            scanf ("%lld", &c);
            sp.root->ch[]->ch[]->lazy += c;
        }
    }
    return ;
}