转移一篇线段树

转载请注明出处，谢谢http://blog.csdn.net/acm_cxlove/article/details/7854526       by---cxlove 

题目：给出一个有序集合，3种操作。插入一个数，删除一个数，都保证序列有序。以及求和

其中求和是将下标%5==3的所有数求和

http://acm.hdu.edu.cn/showproblem.php?pid=4288 

这是一个CF原题，一开始SB写了半天，思路基本对的，5棵线段树，但是维护的时候想复杂了

又学到了离散化的新姿势，离线输入然后离散化。每个位置初始为0

维护sum[5]表示区间内模5的余数分组的和，cnt表示这个区间内有多少个数

[cpp]  view plain copy

1. #include<iostream>  
2. #include<cstdio>  
3. #include<map>  
4. #include<cstring>  
5. #include<cmath>  
6. #include<vector>  
7. #include<queue>  
8. #include<algorithm>  
9. #include<set>  
10. #define inf (1ull<<63)-1  
11. #define N 100005  
12. #define maxn 100005  
13. #define Min(a,b) ((a)<(b)?(a):(b))  
14. #define Max(a,b) ((a)>(b)?(a):(b))  
15. #define pb(a) push_back(a)  
16. #define mem(a,b) memset(a,b,sizeof(a))  
17. #define eps 1e-9  
18. #define zero(a) fabs(a)<eps  
19. #define LL long long  
20. #define ULL unsigned long long  
21. #define lson (step<<1)  
22. #define rson (step<<1|1)  
23. #define MOD 1000000007  
24. #define mp(a,b) make_pair(a,b)  
25. using namespace std;  
26. struct Node{  
27.     int left,right;  
28.     int cnt;  
29.     LL sum[5];  
30. }L[N*4];  
31. int n,tot,a[N],k,q[N];  
32. char str[N][10];  
33. void Bulid(int step,int l,int r){  
34.     L[step].left=l;L[step].right=r;  
35.     L[step].cnt=0;  
36.     memset(L[step].sum,0,sizeof(L[step].sum));  
37.     if(l==r) return ;  
38.     int m=(l+r)/2;  
39.     Bulid(lson,l,m);  
40.     Bulid(rson,m+1,r);  
41. }  
42. void Push_Up(int step){  
43.     for(int i=0;i<5;i++)  
44.         L[step].sum[i]=(LL)L[lson].sum[i]+L[rson].sum[((i-L[lson].cnt)%5+5)%5];  
45. }  
46. void Update(int step,int pos,int num){  
47.     L[step].cnt+=2*k-1;  
48.     if(L[step].left==L[step].right){  
49.         L[step].sum[0]=(LL)k*num;  
50.         return;  
51.     }  
52.     int m=(L[step].left+L[step].right)/2;  
53.     if(pos<=m) Update(lson,pos,num);  
54.     else Update(rson,pos,num);  
55.     Push_Up(step);  
56. }  
57. int main(){  
58.     while(scanf("%d",&n)!=EOF){  
59.         tot=0;  
60.         for(int i=0;i<n;i++){  
61.             scanf("%s",str[i]);  
62.             if(str[i][0]!='s'){  
63.                 scanf("%d",&q[i]);  
64.                 a[tot++]=q[i];  
65.             }  
66.         }  
67.         sort(a,a+tot);  
68.         tot=unique(a,a+tot)-a;  
69.         if(tot==0) memset(L[1].sum,0,sizeof(L[1].sum));  
70.         else  
71.         Bulid(1,1,tot);  
72.         for(int i=0;i<n;i++){  
73.             int pos=lower_bound(a,a+tot,q[i])-a;  
74.             if(str[i][0]=='a')k=1,Update(1,pos,q[i]);  
75.             else if(str[i][0]=='d') k=0,Update(1,pos,q[i]);  
76.             else printf("%I64d\n",L[1].sum[2]);  
77.         }  
78.     }  
79.     return 0;  
80. }  
81.