HDU 4288 Coder [线段树]

　　维护一个可以插入删除的有序序列，每次询问序列中位置mod5=3的数的和。

　　CodeForces原题，因为时限给的太宽，数据太水，STL可以暴力过。

　　用线段树和平衡树都可以做这题，线段树需要先离散化，然后每个区间五个标记分别保存到这个区间左端点的距离mod5=0~4的和，以及一个标记保存这个区间数的个数。合并区间的时候，左儿子因为左端点相同可以直接合并，而右儿子需要推一下，假设左儿子有Num个元素，右儿子本来Mod5=K的点,合并之后到左端点Mod5的值为(num%5+K)%5，所以sum[p][i] = sum[p<<1][i] + sum[p<<1|1][(i-num[p<<1]%5+5) % 5]。

　　换了下代码风格，该加空格的地方加上了空格，尽量让自己的代码写的标准一些吧。

#include <string.h>
#include <stdio.h>
#include <algorithm>
#define calm l + r >> 1;
#define lson l, m, p << 1
#define rson m+1, r, p << 1 | 1
#define MAXN 100005
typedef long long LL;
int n,tot,d[MAXN],unq[MAXN],num[MAXN<<2];
char s[MAXN][10];
LL sum[MAXN<<2][5];
void pushup(int p) {
    for (int i = 0; i < 5; i++)
        sum[p][i] = sum[p<<1][i] + sum[p<<1|1][(i-num[p<<1]%5+5) % 5];
}
void update(int x, int v, int o,int l, int r, int p) {
    num[p] += o ? 1 : -1;
    if (l == r) {
        sum[p][0] = o ? v : 0;
        return;
    }
    int m=calm;
    if(x <= m)update(x, v, o, lson);
    else update(x, v, o, rson);
    pushup(p);
}
int main() {
    //freopen ("test.in" ,"r", stdin);
    while (scanf("%d", &n) != EOF) {
        tot = 0;
        for (int i = 0; i < n; i++) {
            scanf("%s", s[i]);
            if (*s[i] != 's')
                scanf("%d", &d[i]), unq[tot++] = d[i];
        }
        std::sort(unq, unq + tot);
        tot = std::unique(unq, unq + tot) - unq;
        memset(sum, 0, sizeof sum);
        memset(num, 0, sizeof num);
        for (int i = 0; i < n; i++) {
            if (*s[i] == 's') printf("%I64d\n", sum[1][2]);
            else {
                int x = std::lower_bound(unq, unq + tot, d[i]) - unq;
                update(x, d[i], *s[i] == 'a' ? 1: 0, 1, tot ,1);
            }
        }
    }
    return 0;
}