hdu 4288 Coder(线段树)

hdu 4288 Coder

此题表示有一定难度啊、、

首先线段树涉及到增点、删点如果没有办法规避，就只能利用伸展树等其他数据结构了

如果线段树要规避增点、删点的话那一般都要涉及离线操作，此题也是

也是看别人代码理解的思路吧

节点定义一个sum[5]、以及一个cnt表示该节点所代表区间内的数的个数

而怎么去理解sum[5]？

不得不佩服大神算法的巧妙

sum[5]保存的是相对于该节点左端点的MOD 5的值

虽然除最左的儿子节点外，其他节点保存的sum[i]的值并非相对0进行的MOD5操作，但巧妙的是可以在向上更新中用过获取左儿子节点中保存的数的个数一步一步将本节点转换成相对0的MOD5操作

即：

for(int i=0;i<5;i++)
t[p].sum[i]=t[lch].sum[i]+t[rch].sum[((i-t[lch].cnt)%5+5)%5];

在更新的时候只需要维护sum[i]和cnt的值就行了，直接查询t[1].sum[2]便是结果

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<map>
using namespace std;
#define ll __int64
#define MAXN 100005
#define lch p<<1
#define rch p<<1|1
#define mid (t[p].l+t[p].r)>>1
struct node
{
	int l,r;
	ll sum[5];
	int cnt;
}t[MAXN<<2];
void construct(int l,int r,int p)
{
	t[p].l=l,t[p].r=r,t[p].cnt=0;
	memset(t[p].sum,0,sizeof(t[p].sum));
	if(l==r) return ;
	int m=mid;
	construct(l,m,lch);
	construct(m+1,r,rch);
}
void pushup(int p)
{
	t[p].cnt=t[lch].cnt+t[rch].cnt;
	for(int i=0;i<5;i++)
		t[p].sum[i]=t[lch].sum[i]+t[rch].sum[((i-t[lch].cnt)%5+5)%5];
}
void modify(int x,ll val,int p)
{
	if(t[p].l==t[p].r)
	{
		t[p].sum[0]+=val;
		t[p].cnt+=val<0?-1:1;
		return ;
	}
	int m=mid;
	if(x<=m) modify(x,val,lch);
	else modify(x,val,rch);
	pushup(p);
}
int op[MAXN],x[MAXN],xx[MAXN];
map<int,int> rec;
int main()
{
	char oop[5];
	int n;
	while(scanf("%d",&n)!=EOF)
	{
		int i,j;
		for(i=0,j=0;i<n;i++)
		{
			scanf("%s",oop);
			if(oop[0]=='a'||oop[0]=='d')
			{
				op[i]=oop[0]=='a'?1:2;
				scanf("%d",&x[i]);
				xx[j++]=x[i];
			}
			else
				op[i]=3;
		}
		sort(xx,xx+j);
		int up=unique(xx,xx+j)-xx;
		construct(0,up-1,1);
		rec.clear();
		for(int i=0;i<up;i++)
			rec[xx[i]]=i;
		for(int i=0;i<n;i++)
		{
			if(op[i]==1)
				modify(rec[x[i]],(ll)x[i],1);
			else if(op[i]==2)
				modify(rec[x[i]],(ll)(-x[i]),1);
			else
				printf("%I64d\n",t[1].sum[2]);
		}
	}
	return 0;
}