hdu 4288 Coder & CF85-D Sum of Medians （单点更新）

题意：有三种类型的操作，1."add x"表示往集合里添加数x。2.“del x”表示将集合中数x删除。3.“sum”求出从小到大排列的集合中下标模5为3的数的和。集合中的数都是唯一的。

线段树。在线段树中维护当前这个集合中数的个数cnt，和所有的数模5为0……4内的数的和设为mod[0...4]。在进行区间合并的时候，父区间里的mod[0...4]首先等于左子区间里的mod[0...4]，设要加入的右子区间的数为mod[i]，则它应该加到父区间的mod[左子区间的cnt+i]。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <map>
#include <algorithm>
#include <vector>
using namespace std;
typedef long long LL;
#define LL(x) (x<<1)
#define RR(x) (x<<1|1)
const int N=100005;
char str[5];
vector<int> y;
map<int,int> imap;
struct OP
{
	int type,valu;
	void input()
	{
		scanf("%s",str);
		if(strcmp(str,"add")==0) type=0;
		else if(strcmp(str,"del")==0) type=1;
		else type=2;
		if(type==2) return;
		scanf("%d",&valu);
		y.push_back(valu);
	}
}op[N];
struct node
{
	int lft,rht,cnt;
	LL mod[5];
	int mid(){return lft+(rht-lft)/2;}
	void clear(){memset(mod,0,sizeof(mod));}
	void change()
	{
		if(cnt==0) mod[1]=0;
		else mod[1]=y[lft];
	}
};
struct Segtree
{
	node tree[N*4];
	void relax(int ind)
	{
		int c=tree[LL(ind)].cnt;
		for(int i=0;i<5;i++) tree[ind].mod[i]=tree[LL(ind)].mod[i];
		for(int i=0;i<5;i++) tree[ind].mod[(i+c)%5]+=tree[RR(ind)].mod[i];
	}
	void build(int lft,int rht,int ind)
	{
		tree[ind].lft=lft;	tree[ind].rht=rht;
		tree[ind].cnt=0;	tree[ind].clear();
		if(lft!=rht)
		{
			int mid=tree[ind].mid();
			build(lft,mid,LL(ind));
			build(mid+1,rht,RR(ind));
		}
	}
	void updata(int pos,int ind,int valu)
	{
		tree[ind].cnt+=valu;
		if(tree[ind].lft==tree[ind].rht) tree[ind].change();
		else 
		{
			int mid=tree[ind].mid();
			if(pos<=mid) updata(pos,LL(ind),valu);
			else updata(pos,RR(ind),valu);
			relax(ind);
		}
	}
}seg;
int main()
{
	int n;
	while(scanf("%d",&n)!=EOF)
	{
		int sc=0;
		y.clear(); imap.clear();
		for(int i=0;i<n;i++) op[i].input();

		sort(y.begin(),y.end());
		y.erase(unique(y.begin(),y.end()),y.end());
		for(int i=0;i<(int)y.size();i++) imap.insert(make_pair(y[i],sc++));

		seg.build(0,sc,1);
		for(int i=0;i<n;i++)
		{
			if(op[i].type==0) seg.updata(imap[op[i].valu],1,1);
			else if(op[i].type==1) seg.updata(imap[op[i].valu],1,-1);
			else printf("%I64d\n",seg.tree[1].mod[3]);
		}
	}
	return 0;
}