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);
}
}
}