Coder

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

题意：有三种类型的操作，（1）."add x"，表示往集合里添加数x。（2）.“del x”表示将集合中数x删除。（3）.“sum”求出从小到大排列的集合中下标模5为3的数的和。集合中的数都是唯一的。

题解：线段树离线操作，一开始没有想到，看了别人的思路才知道用线段树怎么操作。用5个线段树分别维护：下标取模分别为0 1 2 3 4的数的和。对于一棵树的来说，该区间的sum【u】【i】=sum【u<<1】[i],同时，对于幼儿来说，sum【u<<1|1】【i】应该贡献给父亲sum[u][(i+id)%5],id为左子树的个数。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define lson u<<1
#define  rson u<<1|1
using namespace std;
const int N=100003;
long long sum[4*N][5],a[N],idx[N];
int ll[4*N],rr[4*N],num[4*N],top;
char op[N][20];
int n;
void pushUP(int u){
    num[u]=num[lson]+num[rson];
    int id=num[lson];
    for(int i=0;i<=4;i++){
        sum[u][i]=sum[lson][i];
    }
    for(int i=0;i<=4;i++){
        sum[u][(i+id)%5]+=sum[rson][i];
    }
}
void build(int l,int r,int u){
    ll[u]=l;
    rr[u]=r;
    memset(sum[u],0,sizeof(sum[u]));
    if(l==r){
        num[u]=0;
        return;
    }
    int m=(l+r)/2;
    build(l,m,lson);
    build(m+1,r,rson);
    pushUP(u);
}
void update(int pos,int u,long long val,int fg){
    if(ll[u]==rr[u]){
       sum[u][1]=val*fg;
       num[u]=1*fg;
       return;
    }
    int m=(ll[u]+rr[u])/2;
    if(m>=pos)update(pos,lson,val,fg);
    else
        update(pos,rson,val,fg);
    pushUP(u);
}

int main(){
   while(~scanf("%d",&n)){
         top=0;
       for(int i=1;i<=n;i++){
          scanf("%s",op[i]);
          if(op[i][0]=='a'){
            scanf("%I64d",&a[i]);
            idx[++top]=a[i];
          }
          else if(op[i][0]=='d'){
            scanf("%I64d",&a[i]);
          }
       }
       sort(idx+1,idx+top+1);
       int k=unique(idx+1,idx+top+1)-(idx+1);
         build(1,k,1);
     for(int i=1;i<=n;i++){
          if(op[i][0]=='a'){
           int pos=upper_bound(idx+1,idx+k+1,a[i])-(idx+1);
            update(pos,1,a[i],1);
          }
          else if(op[i][0]=='d'){
           int pos=upper_bound(idx+1,idx+k+1,a[i])-(idx+1);
            update(pos,1,0,0);
          }
          else{
            printf("%I64d
",sum[1][3]);
          }
       }
   }
}