【HDU4288 Coder】离线+离散化+5颗线段树的更新操作

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4288

 

题目大意：有n个操作和一个集合（初始为空），并且集合内元素满足按顺序排列。

现有三种操作：

add x ： 将元素x加入集合中。

del x： 将集合中的x元素删除。

sum： 统计集合中元素下标满足i%5==3的总和。

 

解题思路：  以为在线段树方面小有火候，现在发现自己在线段树方面还没起火，处于冒烟阶段。

                这题让我蛋蛋又碎了，WA了一个晚上，各种调试，最后发现是数组必须开成节点4倍大小，我只开了两倍大小，导致了一个晚上的悲剧（why？个人认为两倍足够了，可以证明的）。

                可以想到五颗线段树的操作，离线处理还是比较难想到的。先离线处理可以知道我们要处理多少个点，进行建树操作。然后接下来对这n个操作进行即时更新，遇见add x，则往x的对应位置进行传递，直到达到叶子节点，更新叶子节点sz为1，del x则更新对应叶子节点sz为0。然后递归回去上传更新，父亲的  sum[u][i]等于左孩子的sum[u][i]，而右孩子上传时则需要处理，因为位置发生了变化，cnt表示父亲的左孩子有cnt个实节点，所以右孩子上传时下标是从cnt+i开始，五种状态随时更新，随时相互转换。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

#define lz 2*u,l,mid
#define rz 2*u+1,mid+1,r
const int maxn=100005;
__int64 sum[4*maxn][6];  ///开成2*maxn WA了一个晚上，蛋蛋都碎了
int a[maxn], X[maxn], sz[4*maxn];
char str[maxn][20];

void push_up(int u)
{
    sz[u]=sz[2*u]+sz[2*u+1];
    for(int i=0; i<5; i++) sum[u][i]=sum[2*u][i];
    for(int i=0; i<5; i++) sum[u][(i+sz[2*u])%5]+=sum[2*u+1][i];
}

void build(int u, int l, int r)
{
    sz[u]=0;
    for(int i=0; i<5; i++) sum[u][i]=0;
    if(l==r) return ;
    int mid=(l+r)>>1;
    build(lz);
    build(rz);
}

void Update(int u, int l, int r, int pos, int c)
{
    if(l==r)
    {
        sz[u]=c;
        sum[u][1]=c*X[l];
        return ;
    }
    int mid=(l+r)>>1;
    if(pos<=mid) Update(lz,pos,c);
    else Update(rz,pos,c);
    push_up(u);
}

int main()
{
    int n;
    while(cin >> n)
    {
        int num=0, pos;
        for(int i=0; i<n; i++)
        {
            scanf("%s",str[i]);
            if(str[i][0]!='s')
            {
                scanf("%d",&a[i]);
                if(str[i][0]=='a') X[++num]=a[i];
            }
        }
        sort(X+1,X+num+1);
        int ep=unique(X+1,X+num+1)-(X+1);  ///去重
        build(1,1,ep);
        for(int i=0; i<n; i++)
        {
            if(str[i][0]=='a') pos=upper_bound(X+1,X+ep+1,a[i])-(X+1), Update(1,1,ep,pos,1);
            else if(str[i][0]=='d') pos=upper_bound(X+1,X+ep+1,a[i])-(X+1), Update(1,1,ep,pos,0);
            else printf("%I64d\n",sum[1][3]);
        }
    }
    return 0;
}