HDU4288 Coder

　　原题传送：http://acm.hdu.edu.cn/showproblem.php?pid=4288

　　线段树。

　　技巧有一下几个：

　　　　1. 先把数据全部输入完毕后再统一查询，进行排序预处理方便计算出元素位置（线段树的堆底是静态的）；

　　　　2. 每个区间节点保存mod 5 = 0...4这5个值；

　　　　3. 合并区间时，左孩子起点就是当前区间起点，可以直接合并，但是相应的右孩子节点变为(k + 5 - num(lson) % 5 ) % 5 （k为合并前mod5的点，num(lson)为左孩子的点数）。

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <iostream>
#define lson (cur << 1)
#define rson (cur << 1 | 1)
#define N 100005
#define LL long long
int v[N], d[N], tot, num[N << 2];
char cmd[N][5];

LL tree[N << 2][5];

void push_up(int cur)
{
    for(int i = 0; i < 5; i ++)
       tree[cur][i] = tree[lson][i] + tree[rson][(i + (5 - num[lson] % 5)) % 5];
}

void update(int cur, int l, int r, int loc, LL k, bool flag)
{
    num[cur] += flag ? 1 : -1;
    if(l == r)
    {
        tree[cur][0] = (flag ? k : 0);
        return ;
    } 
    int mid = (l + r) >> 1;
    if(loc <= mid)
        update(lson, l, mid, loc, k, flag);
    else
        update(rson, mid + 1, r, loc, k, flag);
    push_up(cur);
}

int main()
{
    int i, n;
    while(scanf("%d", &n) != EOF)
    {
        for(tot = i = 0; i < n; i ++)
        {
            scanf("%s", cmd[i]);
            if(cmd[i][0] != 's')
            {
                scanf("%I64d", &d[i]);
                v[tot ++] = d[i];
            }
        }
        std::sort(v, v + tot);
        memset(tree, 0, sizeof tree);
        memset(num, 0, sizeof num);
        for(i = 0; i < n; i++)
        {
            if(cmd[i][0] == 's')
                printf("%I64d\n", tree[1][2]);
            else
            {
                int loc = std::lower_bound(v, v + tot, d[i]) - v;
                update(1, 1, tot, loc, d[i], cmd[i][0] == 'a' ? 1 : 0);
            }
        }
    }
    return 0;
}