hdu 4288(线段树) Coder

首先要离散化！！！学习到了！！看过代码之后就会理解。。

这道题，难在线段树节点的维护。

每个节点即每个区间，cnt[rt]表示该区间的个数，sum[rt][5]表示该区间对5取余相同的数的和。

接下来就是怎么样pushUp(rt)了，即维护该节点：

该节点左孩子有cnt[rt<<1]个数，那么，sum[rt<<1][i]对sum[rt][i]的结果没影响，可以想象一下，直接加上就可以。

那么，sum[rt<<1|1][i]呢？

要求的是对5求余为3的所有数的和。左孩子已经有cnt[rt<<1]个数了，对右孩子有什么影响呢？

可以将cnt[rt<<1]%5，那么可以知道左孩子已经计算完全有5个数，还剩余几个数。那么，右孩子继续接着剩下的数往后累加。

那么，可以知道5-cnt[rt<<1]%5该位置为下一次计算的0位置。

就可以算出sum[rt][i]=sum[rt<<1][i]+sum[rt<<1|1][(5-cnt[rt<<1]%5+i)%5]

 

#include <iostream>
#include<algorithm>
using namespace std;
#define MAXN 100010

int cnt[MAXN<<2];
__int64 sum[MAXN<<2][5];
long num[MAXN];
int flag;

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

void update(int l,int r,int rt,int pos)
{
    cnt[rt]+=flag*2-1;
    if(l==r)
    {
        sum[rt][0]=flag*num[pos];
        return;
    }
    int mid=(l+r)/2;
    if(pos<=mid) update(l,mid,rt<<1,pos);
    else update(mid+1,r,rt<<1|1,pos);
    pushUp(rt);
}

void build(int s,int t,int rt)
{
    cnt[rt]=0;
    for(int i=0;i<5;i++)
        sum[rt][i]=0;
    if(s==t)
        return;
    int mid=(s+t)/2;
    build(s,mid,rt<<1);
    build(mid+1,t,rt<<1|1);
}

int main()
{
    int n;
    char op[MAXN][5];
    int tot;
    long x[MAXN];
    while(scanf("%d",&n)==1)
    {
        tot=0;
        for(int i=0;i<n;i++)
        {
            scanf("%s",&op[i]);
            x[i]=0;
            if(op[i][0]=='a'||op[i][0]=='d')
            {
                scanf("%d",&x[i]);
                num[tot++]=x[i];
            }
        }
        sort(num,num+tot);
        tot=unique(num,num+tot)-num;
        build(0,tot-1,1);
        for(int i=0;i<n;i++)
        {
            int pos;
            if(x[i]) pos=lower_bound(num,num+tot,x[i])-num;
            if(op[i][0]=='a') flag=1,update(0,tot,1,pos);
            else if(op[i][0]=='d') flag=0,update(0,tot,1,pos);
            else printf("%I64d\n",sum[1][2]);
        }
    }
    return 0;
}