HDU 4288 && Codeforces 85D (线段树+离散化+离线处理)

首先，为什么要离散化和离线处理？因为输入的数字会很大，可以达到10^9，但输入的N最多为10^5，即做多添加100000个数(不删除任何数)，所以把输入的值进行排序，离散成1~100000，同时这也可以知道要建多大的线段树。

离散化的时候，经常要用到去重操作和查找操作，一般查找是用二分查找的。通过这题学习到了两个新的STL。

1)unique:去重 2)lower_bound:返回第一个比需要查找的数大的或者相等的数组下标（相当于二分查找）。具体实现在代码中。

最后，这题的pushUp操作是精髓，自己就是想不出来，唉。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define SIZE 100005
#define ls l,mid,rt<<1
#define rs mid+1,r,rt<<1|1

using namespace std;

typedef __int64 Int;
Int sum[SIZE<<2],ans[SIZE<<2][5];
Int arr[SIZE],der[SIZE];
char op[SIZE][5];
int n;

void pushUp(int rt)
{
    for(int i=0; i<5; i++)
        ans[rt][i] = ans[rt<<1][i] + ans[rt<<1|1][(i-sum[rt<<1]%5+5)%5];
}

void build(int l,int r,int rt)
{
    sum[rt] = 0;
    for(int i=0; i<5; i++)ans[rt][i] = 0;
    if(l == r)return;
    int mid = (l + r) >> 1;
    build(ls);
    build(rs);
}

void update(int l,int r,int rt,int pos,int add)
{
    sum[rt] += add;
    if(l == r)
    {
        ans[rt][1] = add==1?der[pos]:0;
        return;
    }
    int mid = (l + r) >> 1;
    if(pos <= mid) update(ls,pos,add);
    else update(rs,pos,add);
    pushUp(rt);
}

int main()
{
    while(~scanf("%d",&n))
    {
        int len = 0;
        for(int i=0; i<n; i++)
        {
            scanf("%s",op[i]);
            if(op[i][0] != 's')
            {
                scanf("%I64d",&arr[i]);
                der[len++] = arr[i];
            }
        }
        if(n == 1)
        {
            if(op[0][0] == 's')
                puts("0");
            continue;
        }
        sort(der,der+len);
        len = unique(der,der+len)-der;
        build(1,len,1);
        int pos;
        for(int i=0; i<n; i++)
        {
            pos = lower_bound(der,der+len,arr[i])-der;
            if(op[i][0] == 's')
                printf("%I64d\n",ans[1][3]);
            else if(op[i][0] == 'a')
                update(1,len,1,pos,1);
            else
                update(1,len,1,pos,-1);
        }
    }
    return 0;
}