hdu 4288 Coder (成都赛区 线段树)

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

题意：

给出一个有序集合，3种操作。插入一个数，删除一个数，都保证序列有序。以及求和

其中求和是将下标%5==3的所有数求和；

 

题解： 线段树   + 离散化 + 离线处理

一开始也是想的 线段树 ，但是 这个和以前的 做过的 一个线段树 不同的 是 ，如果 我们 删除 一个 元素后  ，那么 他的 下标 将会 改变 ，比赛是 不知  如何下手 。。。。。

 

 

同样 是 用 5棵线段树   维护 ，s[0]表示 %5  == 1 的 下标，其他 依次类推  cnt，记录 子树的 元素个数。

想要得到该区间内所有模5等3所有元素的和，左孩子可以求到，两个孩子相互独立，所以求右孩子需要知道（左子树含有 ）多少个元素，因为这样分开求的时候，我们才知道 求右孩子时应该求下标模5等几（)的所有元素的和。

假如我们 求  i ==  3 时 （表示  %5 == 4），对于 左子树 我们 直接 求 就可以 ，但是 对于 右子树，我们 要知道 在 i== 3 的 情况下 ，右子树的 元素的下标 应该 是 %5 == 几？

p[x].s[i] = p[x*2].s[i] + p[x*2+1].s[((i  - p[x*2].cnt)%5 + 5)% 5]；

 

 

#include<cstdio>
#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<set>
#include<map>
#include<queue>
#include<vector>
#include<string>
#define Min(a,b) a<b?a:b
#define Max(a,b) a>b?a:b
#define CL(a,num) memset(a,num,sizeof(a));
#define eps  1e-12
#define inf   10000000
//freopen("data.txt","r",stdin);
const double pi  = acos(-1.0);
typedef   __int64  ll;
const int maxn = 101000 ;
using namespace std;
char c[6];
struct node
{
    int l;
    int r;
    ll s[5] ;
    int cnt ;

}p[maxn*4] ;
ll  a[maxn],b[maxn],tp[maxn] ;
void build(int x,int l,int r)
{
     p[x].cnt = 0;
     p[x].l = l;
     p[x].r = r;
     CL(p[x].s,0) ;
    if(l == r)return ;

    int mid = ( l + r) >> 1;
    build(x*2,l,mid);
    build(x*2+1,mid+1,r) ;
}
void add(int x,int pos,int k)
{
    p[x].cnt += k*2 - 1 ;
    if(p[x].l  == pos && p[x].r == pos)
    {
          p[x].s[0] = k*a[pos] ;//只有 一个元素 所以 % 5 == 1，我们 放到 s[0]里面
          return ;
    }


    int  mid = (p[x].l + p[x].r) >> 1;
    if(pos <=mid )add(x*2,pos,k);
    else add(x*2+1,pos,k);

    for(int i = 0 ; i< 5;i++)
    {
        p[x].s[i] = p[x*2].s[i] + p[x*2+1].s[((i  - p[x*2].cnt)%5 + 5)% 5] ;//实现了 动态的 改变
    }

}
int main()
{
   //freopen("data.txt","r",stdin);
   int n ,i;
   while(scanf("%d",&n)!=EOF)
   {
       CL(tp,-1);
       int num = 0 ;
       for(i = 0 ; i < n;i++)
       {
           scanf("%s",c);
           if(c[0] == 'a')
           {
               tp[i] = 0;
               scanf("%I64d",&b[i]);
               a[num++] = b[i] ;
           }
           if(c[0] == 'd')
           {
               tp[i] = 1;
               scanf("%I64d",&b[i]);
               a[num++] = b[i] ;
           }
           if(c[0] == 's')
           {
               tp[i] = 2 ;
           }
       }
        sort(a,a+num) ;
        int tot = unique(a,a+num) - a ;


        
        build(1,0,tot - 1);

        for(i = 0  ; i< n;i++)
        {
            if(tp[i] == 0)
            {
                add(1,lower_bound(a,a+tot,b[i]) - a ,1);
            }
            if(tp[i] == 1)
            {
                  add(1,lower_bound(a,a+tot,b[i]) - a ,0);
            }
            if(tp[i] == 2)
             printf("%I64d\n",p[1].s[2]);
        }


   }
}