Codeforces-85D Sum of Medians

题目大意：对于一个每个数最多出现一次的序列，可以有add del两种修改和查询三种操作。对于每个查询操作，输出该序列排序后

线段树每个区间分别维护0 1 2 3 4五种余数的位置上数的和。注意这里的位置是相对于每个区间的而非整个序列。另外，还需维护每个区间中当前存在的数字的个数（cnt数组）。这一步的作用是在sum转移时，对于每一个sum[v][i]，从右子树转移时不能直接用sum[rc][i]，而需要减去cnt[lc]%5，将v节点区间内的余数和右子树节点区间内的余数对应上。

每次的更新都是单点修改，只需要将该位置的数字加上或删去，分别对应cnt++;sum+=val和cnt--;sum-=val。对于每次查询输出sum[1][2]即可。

每个数字高达1e9，因此需要离散化+离线。

另注意从下向上的单点修改一定要保证线段树中maxn的大小是2的整数次幂，不然会出锅。

#include<bits/stdc++.h>
using namespace std;

#define pb push_back
#define fi first
#define se second
#define ll long long
#define pq priority_queue
#define pii pair<int,int>
#define mod 998244353
#define debug(x) cerr<<#x<<"="<<x<<'\n'
#define lc v*2
#define rc v*2+1

int n,Q;
const int maxn=(1<<17);
pii q[maxn];
map<int,int> mp;
ll sum[4*maxn][5];
int cnt[4*maxn];

void update(int v) {
	cnt[v]=cnt[lc]+cnt[rc];
	for (int i=0;i<5;i++)
		sum[v][i]=sum[lc][i]+sum[rc][(i-cnt[lc]%5+5)%5];
	return;
}

int main(){
	scanf("%d",&Q);
	for (int i=0;i<Q;i++) {
		string s;
		cin>>s;
		if (s[0]=='s') {
			q[i].fi=2,q[i].se=0;
			continue;
		}
		int x;scanf("%d",&x);
		if (s[0]=='a') q[i]=make_pair(0,x);
		else if (s[0]=='d') q[i]=make_pair(1,x);
		mp[x]=0;
	}
	n=0;
	for (map<int,int>:: iterator it=mp.begin();it!=mp.end();it++) 
		it->second=n++;
	for (int i=0;i<Q;i++) {
		if (q[i].fi==2) cout<<sum[1][2]<<endl;
		else {
			int pos=maxn+mp[q[i].se];
			cnt[pos]=1-q[i].fi;
			sum[pos][0]=cnt[pos]*(ll)q[i].se;
			while (pos>1) {
				pos/=2;
				update(pos);
			}
		}
	}
	return 0;
}