题意:
维护一个有序数列{An},有三种操作:
1、添加一个元素。
2、删除一个元素。
3、求数列中下标%5 = 3的值的和。
解题思路:
看的各种题解,今天终于弄懂了。
由于线段树中不支持添加、删除操作,所以题解写的是用离线做法。
我们来看它是如何解决添加、删除的问题的。
首先将所有出现过的数记录下来,然后排序去重,最后根据去重结果建树,然后每个操作数都会对应线段树中的一个点。
遇到添加、删除操作的时候,只要把那个节点的值改变,然后将它对下标的影响处理好就可以。
那么如何处理这些操作对下标的影响呢?
现在我们考虑一个父区间,假设它的左右子区间已经更新完毕。
显然,左区间中下标%5的情况与父区间中%5的情况完全相同;
可是,右区间中却不一定相同,因为右区间中的下标是以 mid 当作 1 开始的。
那么,只要我们知道左区间中有效元素的个数 cnt,我们就能知道右区间中的下标 i 在父区间中对应的下标为 i+cnt。
所以,虽然我们最终要的只是总区间中下标%5 = 3的和。但是在更新时我们需要知道右区间%5的所有情况。
于是我们要在线段树的每个节点开一个 sum[5] 和一个 cnt,分别记录这个节点中下标%5的5种情况的和与有效元素的个数。
而查询时,直接访问总区间的 sum[3] 即可。
如此,题目便可解了。复杂度O(M logN logN)。
以前用的是线段树,这次用暴力来了一发
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<map>
using namespace std;
#define MOD 1000000007
const int INF=0x3f3f3f3f;
const double eps=1e-;
typedef long long ll;
#define cl(a) memset(a,0,sizeof(a))
#define ts printf("*****\n");
const int N=;
int n,m,tt;
int main()
{
int i,j,k,ca=;
#ifndef ONLINE_JUDGE
freopen("1.in","r",stdin);
#endif
while(scanf("%d",&n)!=EOF)
{
int len=;
vector<int> vc;
int x;
vector<int>::iterator it;
char s[];
while(n--)
{
scanf("%s",s);
if(s[]=='a')
{
len++;
scanf("%d",&x);
it=lower_bound(vc.begin(),vc.end(),x);
vc.insert(it,x);
}
else if(s[]=='d')
{
len--;
scanf("%d",&x);
it=lower_bound(vc.begin(),vc.end(),x);
vc.erase(it);
}
else
{
ll sum=;
for(i=;i<len;i+=)
{
sum+=vc[i];
}
printf("%I64d\n",sum);
}
}
}
}
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define root 1,n,1
#define mid ((l+r)>>1)
#define ll long long
#define cl(a) memset(a,0,sizeof(a))
#define ts printf("*****\n");
using namespace std;
const int MAXN=;
ll sum[MAXN<<][];
vector<int> num;
int cnt[MAXN<<],val[MAXN];
char cz[MAXN],s[];
int n,m,t;
int getid(int v)
{
return lower_bound(num.begin(),num.end(),v)-num.begin()+;
}
void pushup(int rt)
{
cnt[rt]=cnt[rt<<]+cnt[rt<<|];
for(int i=;i<;i++) sum[rt][i]=sum[rt<<][i];
for(int i=;i<;i++) sum[rt][(i+cnt[rt<<])%]+=sum[rt<<|][i]; //这两个不能放一起,找了好长时间
}
int tot=;
void update(int pos,int v,int l,int r,int rt)
{
if(l==r)
{
sum[rt][]=num[pos-]*v;
cnt[rt]=v;
return;
}
if(pos<=mid) update(pos,v,lson);
else update(pos,v,rson);
pushup(rt);
}
int main()
{
int i,j,k,q;
#ifndef ONLINE_JUDGE
freopen("1.in","r",stdin);
#endif
while(~scanf("%d",&n))
{
num.clear();
for(i=;i<=n;i++)
{
scanf("%s",s);
cz[i]=s[];
if(s[]!='s')
{
scanf("%d",val+i);
num.push_back(val[i]);
}
}
cl(sum),cl(cnt);
sort(num.begin(),num.end());
num.erase(unique(num.begin(),num.end()),num.end());
int sumn=num.size();
for(int i=;i<=n;i++)
{
if(cz[i] == 'a') update(getid(val[i]),,,sumn,);
else if(cz[i] == 'd') update(getid(val[i]),,,sumn,);
else printf("%I64d\n",sum[][]);
}
}
return ;
}