【codeforces 85D】Sum of Medians 线段树单点更新+离散化

题目链接：http://codeforces.com/contest/85/problem/D
题目大意：有n次操作，操作分3种类型，add x、del x、sum。
add x：向数组中插入数字x，保证插入之前数组中不存在x。
del x：从数组中删除数字x，保证删除之前数组中一定存在x。
sum：将数组从小到大排序后，计算所有第i%5==3（也就是3、8、13、18这样的位置）位的数字之和并输出。

这道题我的第一想法是：是否可以通过建一个树的方式，求出每个区间内i%5==3的位置的数字和？但是想了一下发现没办法更新，因为在前面插入一个数字之后，后面的位置就都变化了，并且无法知道变化后的数字是什么。

后来看了几位大神的题解，才明白：可以让线段树的每个点都维护5个值，分别是在这段区间内i%5==0、1、2、3、4的数字的个数。

这样的好处是显而易见的，这种情况下，如果再在前面插入数字，后面的数字只要右移一位就可以了。比如原先有一个数字x是在第5位，也就是i%5==0的位置，这时候插入一个小于x的数y，那么x就变成了第6位，即i%5==1的位置，就可以进行计算了。

在计算过程中还有一点需要仔细想一下：在将儿子节点存储的数据上传的时候，该如何计算呢？

首先，左儿子的计算很简单，因为它没有受到任何影响，一个数在左儿子的区间内是第几位，它在父亲节点的区间内还是第几位，因此左儿子可以直接按照对应位置加上去。

而右儿子，因为左儿子中还存在一些数字，会将右儿子在父亲节点中的位置向右推，那么我们就需要对每个节点多定义一个变量num，来计数这个区间中有多少个数字，这样就可以知道左儿子中有几个数了。于是现在，向上更新的写法也就呼之欲出了：

树的定义：

struct node1//定义树
{
    int l,r;//左右区间
    int num;//记录这段区间内的数字格数
    __int64 sum[10];//记录区间内i%5==0.1.2.3.4的数字个数
}tree[400100];

树中数据的上传：

void push_up(int rt)
{
    tree[rt].num=tree[rt*2].num+tree[rt*2+1].num;
    //父亲节点的数字个数等于两个儿子的和
    for(int i=0;i<5;i++)
    {
        int t=(100000+i-tree[rt*2].num)%5;
        //t计算的是父亲节点中的这一位该接收的是
        //右儿子中哪一位的值
        tree[rt].sum[i]=tree[rt*2].sum[i]+tree[rt*2+1].sum[t];
        //更新父亲节点的sum数组
    }
    return ;
}

解决了这个关键的问题，其他的部分也就迎刃而解了：
先存储add x中的所有x值，一共m个，对这m个数排序、建树，这里可以不用存储del x中的x值，因为del x中出现的x在add x中肯定出现过。（离散化）
然后对于add和del操作，单点更新，在树中查找x所在的位置，改变那个位置的相关内容，并上传。
对于sum操作，直接输出tree[1].sum[3]就可以了。

AC代码：（该注释的都注释在上面了）

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
using namespace std;
struct node
{
    char c[10];
    int x;
}k[100100];
__int64 x[100100];
struct node1
{
    int l,r;
    int num;
    __int64 sum[10];
}tree[400100];
void build_tree(int rt,int l,int r)
{
    tree[rt].l=l;
    tree[rt].r=r;
    tree[rt].num=0;
    memset(tree[rt].sum,0,sizeof(tree[rt].sum));
    if(l==r)return ;
    int mid=(l+r)/2;
    build_tree(rt*2,l,mid);
    build_tree(rt*2+1,mid+1,r);
    return ;
}
void push_up(int rt)
{
    tree[rt].num=tree[rt*2].num+tree[rt*2+1].num;
    for(int i=0;i<5;i++)
    {
        int t=(100000+i-tree[rt*2].num)%5;
        tree[rt].sum[i]=tree[rt*2].sum[i]+tree[rt*2+1].sum[t];
    }
    return ;
}
void update_tree(int rt,int l,int flag)
{
    if(tree[rt].l==tree[rt].r)
    {
        tree[rt].num+=flag;
        tree[rt].sum[1]+=(flag*x[l]);
        return ;
    }
    int mid=(tree[rt].l+tree[rt].r)/2;
    if(l<=mid)update_tree(rt*2,l,flag);
    else if(l>=mid+1)update_tree(rt*2+1,l,flag);
    push_up(rt);
    return ;
}
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        int m=1;
        for(int i=1;i<=n;i++)
        {
            scanf("%s",k[i].c);
            if(k[i].c[0]=='a')
            {
                scanf("%I64d",&k[i].x);
                x[m++]=k[i].x;
            }
            else if(k[i].c[0]=='d')
            {
                scanf("%I64d",&k[i].x);
            }
        }
        m--;
        if(m==0)
        {
            build_tree(1,1,1);
        }
        else
        {
            sort(x+1,x+m+1);
            build_tree(1,1,m);
        }
        for(int i=1;i<=n;i++)
        {
            int l=lower_bound(x+1,x+m+1,k[i].x)-x;
            if(k[i].c[0]=='a')
            {
                update_tree(1,l,1);
            }
            else if(k[i].c[0]=='d')
            {
                update_tree(1,l,-1);
            }
            else if(k[i].c[0]=='s')
            {
                printf("%I64d\n",tree[1].sum[3]);
            }
        }
    }
    return 0;
}