http://acm.hdu.edu.cn/showproblem.php?pid=4288
初始有一个空集合,有N个操作,1 add x向集合中加一个元素x,2 del x在集合中删除一个元素x,sum询问下标mod 5为3的元素的和。
建立一棵线段树,能够实现数组中元素的删除和添加,维护区间的和。
重点是怎么在元素随时变动的条件下求下标mod 5 为3的数的和。首先节点有一个信息cnt表示区间内元素的个数,sum[5]
代表了在这个区间中下表分别对5取余的元素的和,注意是在当前区间内,若当前有一个元素那么存入sum[1]中。然后就是push_up时,sum【】的维护,这个挺难想的,看了题解然后和同学讨论了好久才明白。
#include <stdio.h>
#include <iostream>
#include <map>
#include <set>
#include <list>
#include <stack>
#include <vector>
#include <math.h>
#include <string.h>
#include <queue>
#include <string>
#include <stdlib.h>
#include <algorithm>
//#define LL long long
#define LL __int64
#define eps 1e-12
#define PI acos(-1.0)
using namespace std;
const int INF = 0x3f3f3f3f;
const int maxn = 100010;
struct Info
{
char str[5];
LL num;
}info[maxn];
LL x[maxn];
struct node
{
int l,r;
int cnt;
LL sum[6];
}tree[maxn*4];
void build(int v, int l, int r)
{
tree[v].l = l;
tree[v].r = r;
tree[v].cnt = 0;
tree[v].sum[0] = tree[v].sum[1] = tree[v].sum[2] = tree[v].sum[3] = tree[v].sum[4] = 0;
if(l == r)
return;
int mid = (l+r)>>1;
build(v*2,l,mid);
build(v*2+1,mid+1,r);
}
int Binsearch(int l, int r, int key)
{
int mid,low = l,high = r;
while(high >= low)
{
mid = (low + high) >> 1;
if(x[mid] == key)
return mid;
if(x[mid] > key)
high = mid-1;
else low = mid+1;
}
return -1;
}
void update(int v, int pos, LL num, int f)
{
tree[v].cnt += f;
if(tree[v].l == tree[v].r)
{
tree[v].sum[1] += num;
return;
}
int mid = (tree[v].l + tree[v].r) >> 1;
if(pos <= mid)
update(v*2,pos,num,f);
else update(v*2+1,pos,num,f);
//重点,维护区间的和。
for(int i = 0; i < 5; i++)
tree[v].sum[i] = tree[v*2].sum[i] + tree[v*2+1].sum[((i-tree[v*2].cnt)%5+5)%5];
}
int main()
{
int n;
int cnt;
while(~scanf("%d",&n))
{
cnt = 0;
for(int i = 1; i <= n; i++)
{
scanf("%s",info[i].str);
if(strcmp(info[i].str,"sum") != 0)
{
scanf("%I64d",&info[i].num);
x[++cnt] = info[i].num;
}
}
//离散化
sort(x+1,x+1+cnt);
cnt = unique(x+1,x+1+cnt)-(x+1);
build(1,1,cnt);
int pos;
for(int i = 1; i <= n; i++)
{
if(info[i].str[0] != 's')
pos = Binsearch(1,cnt,info[i].num);//二分寻找下标
if(info[i].str[0] == 'a')
{
update(1,pos,info[i].num,1);
}
else if(info[i].str[0] == 'd')
{
update(1,pos,-info[i].num,-1);
}
else
printf("%I64d\n",tree[1].sum[3]);
}
}
return 0;
}