hdu 4288 线段树+离散化+离线处理

http://acm.hdu.edu.cn/showproblem.PHP?pid=4288

正确做法：
1、节点内维护sum[0...4]分别代表区间内%5==i的和

2、结点维护点的个数，cnt

3、离散化处理，然后每次插入时，经过的结点cnt+1或-1，叶子节点Sum[0]就是节点值，父节点的sum这样维护了：很不错的方法

void pushup(int rt)  
{  
    for(int i=0;i<5;i++)  
        nodes[rt].sum[i]=nodes[ls(rt)].sum[i]+nodes[rs(rt)].sum[ ((i-nodes[ls(rt)].cnt)%5+5)%5 ];  
}  

4、离散化+离线，可以使删除数比较方便，不用map

新学会的离散化方法：

for(int i=0;i<n;i++)  
{  
    scanf("%d",&a[i]);  
    tmp[num++]=a[i];  
}  
sort(tmp,tmp+num);  
num=unique(tmp,tmp+num)-tmp;  
//然后可以这样查询位置  
int pos=lower_bound(tmp,tmp+num,a[i])-tmp+1; 

代码：

#include <iostream>
#include<algorithm>
#include<stdio.h>
#include<string.h>
#include<math.h>
#define lson node<<1,l,mid
#define rson node<<1|1,mid+1,r
using namespace std;
typedef long long ll;
const int maxn=1e5+100;
char op[maxn][6];
int a[maxn], x[maxn];
ll tree[maxn<<2][5];
int cnt[maxn<<2];

void pushup(int node)
{
    for(int i=0;i<5;i++)
    {
        tree[node][i]=tree[node<<1][i]+tree[node<<1|1][(i-cnt[node<<1]%5+5)%5];
    }
}

void updata(int node, int l,int r,int pos,int x)
{
    x>0? ++cnt[node]:--cnt[node];
    if(l==r)
    {
        tree[node][0]=x;
        return ;
    }
    int mid=(l+r)>>1;
    if(mid>=pos) updata(lson,pos,x);
    else updata(rson,pos,x);
    pushup(node);
}


int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {

        memset(tree,0,sizeof(tree));
        memset(cnt,0,sizeof(cnt));
        int num=0;
        for(int i=0;i<n;i++)
        {
            scanf("%s",op[i]);
            if(op[i][0]!='s')
            {
                scanf("%d",&a[i]);
                x[num++]=a[i];
            }
        }
        sort(x,x+num);
        num=unique(x,x+num)-x;

        for(int i=0;i<n;i++)
        {
            if(op[i][0]=='s') printf("%I64d\n",tree[1][2]);
            else
            {
                 int pos=(int)(lower_bound(x,x+num,a[i])-x)+1;
                 if(op[i][0]=='a') updata(1,1,num,pos,a[i]);
                 else   updata(1,1,num,pos,0);
            }
        }
    }
    return 0;
}