题意:给你一些操作,给集合里加数字,给集合里删除数字,求和是对从x开始的,间隔L的数字的和,数字放进集合内就被自动排为升序,对于每次求和输出结果
题解:因为数字很大所以需要离散化,先离线离散化,然后就是直接更新,但是因为要间隔求和,并且集合里数字的个数会变化,所以需要在每个节点维护L个值,这L个值是在这个节点(包含一些升序的数),从x位开始的节点内的和,怎么理解呢,如果这个节点有数字1 2 3 4 5 ,间隔为2,sum[rt][0]=1+3+5,sum[rt][1]=2+4,这样的意思
所以每次更新的时候,就是直接把更新的值修改在单点sum[rt][0]上即可,然后就是一段节点的sum怎么更新,sum[rt][i],显然在左儿子里面,需要取sum[lrt][i],为什么呢,因为rt这个节点和lrt这个节点的开头是一起的,所以rt节点从i开始,等于lrt节点从i开始,然后就是rrt节点里应该从哪一位开始加,lrt里有一些数字,然后取了i,i+L。。。,看最后有几个没被取,然后L-这个值就是rrt里面的开头,怎么求这个开头呢,要记录lrt节点里有多少个值num[lrt],然后假设从求i位的和,((num[lrt]-(i+1))%L+L)%L,这个就是说明lrt节点最后有几个没被选,然后L-((num[lrt]-(i+1))%L+L)%L -1就是在rrt里面的开头位置(位置从0到L-1),这样就OK了啊;
可能说的有点抽象,所以需要画图好好分析。
#include <map> #include <set> #include <stack> #include <queue> #include <cmath> #include <string> #include <vector> #include <cstdio> #include <cctype> #include <cstring> #include <sstream> #include <cstdlib> #include <iostream> #include <algorithm> #pragma comment(linker,"/STACK:102400000,102400000") using namespace std; #define MAX 100005 #define MAXN 500005 #define maxnode 105 #define sigma_size 2 #define lson l,m,rt<<1 #define rson m+1,r,rt<<1|1 #define lrt rt<<1 #define rrt rt<<1|1 #define middle int m=(r+l)>>1 #define LL long long #define ull unsigned long long #define mem(x,v) memset(x,v,sizeof(x)) #define lowbit(x) (x&-x) #define pii pair<int,int> #define bits(a) __builtin_popcount(a) #define mk make_pair #define limit 10000 //const int prime = 999983; const int INF = 0x3f3f3f3f; const LL INFF = 0x3f3f; const double pi = acos(-1.0); const double inf = 1e18; const double eps = 1e-9; const LL mod = 1e9+7; const ull mx = 1e9+7; /*****************************************************/ inline void RI(int &x) { char c; while((c=getchar())<'0' || c>'9'); x=c-'0'; while((c=getchar())>='0' && c<='9') x=(x<<3)+(x<<1)+c-'0'; } /*****************************************************/ char s[MAX][10]; int a[MAX]; int b[MAX]; LL sum[MAX<<2][5]; int numb[MAX<<2]; int n,L; void pushup(int rt){ numb[rt]=numb[lrt]+numb[rrt]; for(int i=0;i<L;i++){ sum[rt][i]=sum[lrt][i]+sum[rrt][L-((numb[lrt]-(i+1))%L+L)%L-1]; } } void build(int l,int r,int rt){ numb[rt]=0; for(int i=0;i<L;i++) sum[rt][i]=0; if(l==r) return; middle; build(lson); build(rson); } void update(int l,int r,int rt,int pos,int d){ if(l==r){ sum[rt][0]+=d; if(sum[rt][0]) numb[rt]++; else numb[rt]--; return ; } middle; if(pos<=m) update(lson,pos,d); else update(rson,pos,d); pushup(rt); } int main(){ //freopen("test.txt","r",stdin); int kase=0; while(cin>>n>>L){ int num=0; kase++; for(int i=0;i<n;i++){ scanf("%s%d",s[i],&a[i]); if(s[i][0]!='s') b[num++]=a[i]; } sort(b,b+num); int tot=unique(b,b+num)-b; build(1,n,1); printf("Case %d:\n",kase); for(int i=0;i<n;i++){ if(s[i][0]=='s') printf("%lld\n",sum[1][a[i]-1]); else{ int pos=lower_bound(b,b+num,a[i])-b; if(s[i][0]=='d') a[i]=-a[i]; update(1,n,1,pos+1,a[i]); } } } return 0; }