HDU-4288：Coder（线段树+离线操作）

题目链接：点击打开链接

题目大意：

给你一些数让你依次添加到一个集合中，集合中的数是递增的。并且中间会求集合中下标mod5=3的数的和

解题思路：

自己完全不会做，表示线段树怎么能用来写这种东西。

后来还是给大神跪了。据说可以暴力压时间水过，不过也没尝试。

具体的思路就是先将所有数离线保存起来，然后依次再加入线段树中，具体操作也说不太清楚。

中间有一步左加右减的操作。很神奇。先将代码搬过来 t[rt].sum[i]=t[lson].sum[i]+t[rson].sum[(i-t[lson].cnt%5+5)%5];  cnt表示当前节点有效数字

可以看到当前节点 mod5=i 的值是由左节点 mod5=i 的值+右节点 mod5=x（暂且用x表示）。x=(i-t[lson].cnt%5+5)%5

这部分可能有些难以理解，举个例子说明吧   假设现在左节点有4个有效数字 ，右节点有4个有效数字。  那么他们在集合中的下标理论上是 1到8.

假设现在求 mod5=3 的值。那么左儿子就要取 下标mod5=3 的那些数，左儿子下标是正常从1开始的。但是右儿子的有效数字虽然集合中的下标是从5到8，但是在它的线段树中实际上还是1到4，那么我们当前节点mod5=3的数应该有两个 下标为3和下标为8。但是右儿子现在下标不是从5开始，所以就需要转化一下即  x=(i-t[lson].cnt%5+5)%5

下面贴代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <vector>
#include <queue>
#include <map>
#include <algorithm>
#include <set>
#include <functional>
#define rank ra
#define lson rt<<1
#define rson rt<<1|1
#define pb push_back
using namespace std;
typedef long long ll;
int n,m,ans;
int a[100010],v[100010];
char c[100010][5];
struct node
{
    int l,r,mid;
    int cnt;        //记录有效数字的个数
    ll sum[5];      //分别mod5=i的和
}t[500000];
void pushup(int rt)
{
    t[rt].cnt=t[lson].cnt+t[rson].cnt;  //更新有效数字
    for(int i=0;i<5;i++)        //更新当前值得和
        t[rt].sum[i]=t[lson].sum[i]+t[rson].sum[(i-t[lson].cnt%5+5)%5];
}
void build(int l,int r,int rt)  //初始化
{
    int mid=(l+r)>>1;
    t[rt].l=l;t[rt].r=r;
    t[rt].mid=mid;
    for(int i=0;i<5;i++)
        t[rt].sum[i]=0;
    t[rt].cnt=0;
    if(l==r)
        return ;
    build(l,mid,lson);
    build(mid+1,r,rson);
}
void update(int idx,int flag,int rt)
{
    if(t[rt].l==t[rt].r)
    {
        if(flag)        //更新操作
        {
            t[rt].sum[1]=v[idx];
            t[rt].cnt=1;
        }
        if(flag<0)
        {
            t[rt].sum[1]=0;
            t[rt].cnt=0;
        }
        return ;
    }
    if(idx<=t[rt].mid)
        update(idx,flag,lson);
    if(idx>t[rt].mid)
        update(idx,flag,rson);
    pushup(rt);
}
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        int p=0;
        for(int i=1;i<=n;i++)
        {
            scanf(" %s",c[i]);
            if(c[i][0]!='s')
            {
                scanf("%d",&a[i]);
                v[++p]=a[i];        //离线储存
            }
        }
        int k=1;
        sort(v+1,v+1+p);        //排序去重
        for(int i=2;i<=p;i++)
        {
            if(v[i]!=v[i-1])
                v[++k]=v[i];
        }
        build(1,k,1);
        for(int i=1;i<=n;i++)
        {
            if(c[i][0]=='s')
                cout<<t[1].sum[3]<<endl;
            int idx=lower_bound(v+1,v+1+k,a[i])-v;      //找到大于等于它的最小的数所在位置
            if(c[i][0]=='a')
                update(idx,1,1);
            if(c[i][0]=='d')
                update(idx,-1,1);
        }
    }
}