hdu 4288 Coder
http://acm.hdu.edu.cn/showproblem.php?pid=4288
题意:add 就是在集合里面加上一个数x; del 就是从集合里删去一个数x; sum是求位置i%5==3的数的和。
tree[i].sum[5] 里面数组存的是不同位置%5之后分别对应的和。
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
1 #include <cstdio> 2 #include <cstring> 3 #include <algorithm> 4 #define maxn 100010 5 using namespace std; 6 7 char str[maxn][4]; 8 int x[maxn],xx[maxn]; 9 struct node 10 { 11 int r,l; 12 __int64 sum[6]; 13 int cnt; 14 }tree[maxn*4]; 15 16 void build(int i,int l,int r) 17 { 18 tree[i].l=l;tree[i].r=r; 19 for(int j=0; j<5; j++) tree[i].sum[j]=0; 20 tree[i].cnt=0; 21 if(l==r) return ; 22 int mid=(l+r)>>1; 23 build(i<<1,l,mid); 24 build(i<<1|1,mid+1,r); 25 } 26 27 void update(int i,int l,int r,int c,int num) 28 { 29 tree[i].cnt+=num; 30 if(tree[i].l==l&&tree[i].r==r) 31 { 32 tree[i].sum[0]+=c; 33 return ; 34 } 35 int mid=(tree[i].l+tree[i].r)>>1; 36 if(r<=mid) 37 { 38 update(i<<1,l,r,c,num); 39 } 40 else if(l>mid) 41 { 42 update(i<<1|1,l,r,c,num); 43 } 44 else 45 { 46 update(i<<1,l,mid,c,num); 47 update(i<<1|1,mid+1,r,c,num); 48 } 49 for(int j=0; j<5; j++) 50 { 51 tree[i].sum[j]=tree[i<<1].sum[j]+tree[i<<1|1].sum[((j-tree[i<<1].cnt)%5+5)%5]; 52 } 53 } 54 int bs(int l,int r,int data) 55 { 56 while(l<=r) 57 { 58 int mid=(l+r)>>1; 59 if(xx[mid]==data) 60 { 61 return mid; 62 } 63 else if(xx[mid]>data) 64 { 65 r=mid-1; 66 } 67 else 68 l=mid+1; 69 } 70 } 71 int main() 72 { 73 int n; 74 while(scanf("%d",&n)!=EOF) 75 { 76 int t1=0; 77 for(int i=0; i<n; i++) 78 { 79 scanf("%s",str[i]); 80 if(strcmp(str[i],"sum")) 81 { 82 scanf("%d",&x[i]); 83 xx[t1++]=x[i]; 84 } 85 } 86 sort(xx,xx+t1); 87 t1=unique(xx,xx+t1)-xx; 88 build(1,1,t1); 89 for(int i=0; i<n; i++) 90 { 91 int l1=bs(0,t1-1,x[i]); 92 if(!strcmp(str[i],"add")) 93 { 94 update(1,l1+1,l1+1,x[i],1); 95 } 96 else if(!strcmp(str[i],"del")) 97 { 98 update(1,l1+1,l1+1,-x[i],-1); 99 } 100 else if(!strcmp(str[i],"sum")) 101 { 102 printf("%I64d\n",tree[1].sum[2]); 103 } 104 } 105 } 106 return 0; 107 }