1500字范文,内容丰富有趣,写作好帮手!
1500字范文 > Codeforces-85D Sum of Medians

Codeforces-85D Sum of Medians

时间:2019-08-29 15:20:03

相关推荐

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+1int 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;}

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。