题意:对一个集合进行插入与删除操作。要求询问某个时刻,集合中的元素从小到大排序之后,序号%5 ==3 的元素值之和。
首先元素的值可以达到10^9 所以,首先离散化,将所有可能的元素值映射到正整数。
然后线段树的话,用index 存当前节点 所含的元素数量。
用 D [R] 存 所含的元素中,序号%5 ==R 的元素值之和。
则可以用左右子树的这些信息来求出本节点的这些信息,具体见代码。
#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <map>
#define maxn 100010
using namespace std;
//记录操作
int n;
int op[maxn];
//离散化
int cnt;
struct A{
int index;
int value;
void Set(int I,int V){index=I;value=V;}
bool operator<(const A&B)const{return value<B.value;}
}h[maxn];
map<int,int>MP;
//线段树
struct Node{
int index;
long long D[5];
void clear(){index=0;memset(D,0,sizeof(D));}
void show(){
printf("index:%d \n%d %d %d %d %d \n",index,D[0],D[1],D[2],D[3],D[4]);
}
}H[maxn<<2];
void PushUp(int rt){
H[rt].index=H[rt<<1].index+H[rt<<1|1].index;
for(int r=0;r<5;++r){
H[rt].D[r]=H[rt<<1].D[r]+H[rt<<1|1].D[(r+5-(H[rt<<1].index%5))%5];
}
}
void build(int l,int r,int rt){
if(l==r){
H[rt].clear();
return;
}
int m=(l+r)>>1;
build(l,m,rt<<1);
build(m+1,r,rt<<1|1);
PushUp(rt);
}
void Add(int X,int C,int l,int r,int rt){
if(l==r){
H[rt].clear();
H[rt].index=1;
H[rt].D[1]=C;
return;
}
int m=(l+r)>>1;
if(X <= m) Add(X,C,l,m,rt<<1);
if(X > m) Add(X,C,m+1,r,rt<<1|1);
PushUp(rt);
}
void Del(int X,int l,int r,int rt){
if(l==r){
H[rt].clear();
return;
}
int m=(l+r)>>1;
if(X <= m) Del(X,l,m,rt<<1);
if(X > m) Del(X,m+1,r,rt<<1|1);
PushUp(rt);
}
int main(void)
{
while(~scanf("%d",&n)){
cnt=0;MP.clear();
//记录操作
for(int i=0;i<n;++i){
char Op[5];
scanf("%s",Op);
switch(Op[0]){
case 'a':
scanf("%d",&op[i]);
if(!MP.count(op[i])){
++cnt;
h[cnt].Set(cnt,op[i]);
MP[op[i]]=1;
}
break;
case 'd':scanf("%d",&op[i]);op[i]=-op[i];break;
case 's':op[i]=0;break;
default:break;
}
}
//离散化
sort(h+1,h+cnt+1);
for(int i=1;i<=cnt;++i){
MP[h[i].value]=i;
}
build(1,cnt,1);
//线段树
for(int i=0;i<n;++i){
if(op[i]){
if(op[i]>0){
Add(MP[op[i]],op[i],1,cnt,1);
}
else{
Del(MP[-op[i]],1,cnt,1);
}
}
else{
cout<<H[1].D[3]<<endl;
}
}
}
return 0;
}