CodeForces 85 D.Sum of Medians（线段树）

Description
定义一个k个元素的序列S={a[1],a[2],…,a[k]},a[i] < a[I+1]的值为，初始S为空，n种操作，操作有三种：
add x：将x加入到集合S中
del x：将x从集合S中删除
sum：查询集合S的值
Input
第一行一整数n表示操作数，之后n行每行一个操作(1<=n<=1e5,x<=1e9)
Output
对于每次sum操作，输出一个结果
Sample Input
6
add 4
add 5
add 1
add 2
add 3
sum
Sample Output
3
Solution
离线输入所有操作，对操作数离散化建线段树，每个节点维护五个值表示该区间中%5=0,1,2,3,4的位置的数之和，以及该区间中的数据个数，那么每次区间合并时，要计算合并后区间的%5=i的位置的数之和Sum[t][i]，设左子树数据个数为num，那么有Sum[t][i]=Sum[lson][i]+Sum[rson][((i-num)%5+5)%5]，每次sum操作只需要输出Sum[1][2]即可
Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 111111
#define ls (t<<1)
#define rs ((t<<1)|1)
int Size[maxn<<2];
ll Sum[maxn<<2][5];
int n,m,x[maxn],val[maxn];
map<int,int>id;
char s[maxn],op[5];
set<int>S;
set<int>::iterator it;
void push_up(int t)
{
    Size[t]=Size[ls]+Size[rs];
    for(int i=0;i<5;i++)Sum[t][i]=Sum[ls][i]+Sum[rs][((i-Size[ls])%5+5)%5];
}
void update(int state,int v,int l,int r,int t)
{
    if(l==r)
    {
        Size[t]+=state;
        for(int i=0;i<5;i++)Sum[t][i]=0;
        if(Size[t])Sum[t][0]=val[l];
        return ;
    }
    int mid=(l+r)/2;
    if(v<=mid)update(state,v,l,mid,ls);
    else update(state,v,mid+1,r,rs);
    push_up(t);
}
int main()
{
    while(~scanf("%d",&n))
    {
        S.clear();
        for(int i=1;i<=n;i++)
        {
            scanf("%s",op);
            s[i]=op[0];
            if(s[i]!='s')
            {
                scanf("%d",&x[i]);
                S.insert(x[i]);
            }
        }
        m=S.size();
        it=S.begin();
        for(int i=1;i<=m;i++,it++)
            val[i]=*it,id[*it]=i;
        memset(Sum,0,sizeof(Sum));
        memset(Size,0,sizeof(Size));
        for(int i=1;i<=n;i++)
        {
            if(s[i]=='s')printf("%I64d\n",Sum[1][2]);
            else if(s[i]=='a')update(1,id[x[i]],1,m,1);
            else update(-1,id[x[i]],1,m,1);
        }
    }
    return 0;
}