标签：
Description
Input
Output
Sample Input
1 10 1 2 3 4 5 6 7 8 9 10 Query 1 3 Add 3 6 Query 2 7 Sub 10 2 Add 6 3 Query 3 10 End
Sample Output
Case 1: 6 33 59
#include<iostream> #include<stdio.h> #include<math.h> using namespace std; const int maxx = 50005; int tree[maxx<<2]; int a[maxx]; int n; void build(int root,int l,int r) { //cout<<"root:"<<root<<" l,r:"<<l<<" "<<r<<endl; if(l==r) { tree[root]=a[l]; } else { int mid=(l+r)>>1; build(root<<1,l,mid); build((root<<1)+1,mid+1,r); tree[root]=tree[root<<1]+tree[(root<<1)+1]; } } void update(int root,int l,int r,int pos,int val) { if(pos==l&&pos==r) { tree[root]+=val; return; } int mid=(l+r)>>1; if(pos<=mid) update(root<<1,l,mid,pos,val); else update((root<<1)+1,mid+1,r,pos,val); tree[root]=tree[root<<1]+tree[(root<<1)+1]; } int query(int root,int l,int r,int ql,int qr) { //cout<<"root:"<<root<<" l,r:"<<l<<" "<<r<<" ql,qr "<<ql<<" "<<qr<<" tree[root] is :"<<tree[root]<<endl; if(l==ql&&qr==r) { return tree[root]; } int mid=(l+r)>>1; if(qr<=mid) { return query(root<<1,l,mid,ql,qr); } else if(ql>mid) { return query((root<<1)+1,mid+1,r,ql,qr); } else return (query(root<<1,l,mid,ql,mid)+query((root<<1)+1,mid+1,r,mid+1,qr)); } void print() { int j=1,t=1; for(int i=1; i<=(n<<2); i++) { j++; printf("%d ",tree[i]); if(j>pow(2,(t-1))) { printf("\n"); j=1; t++; } } printf("\n\n"); } int main() { int t; int cas=1; scanf("%d",&t); while(t--) { printf("Case %d:\n",cas++); scanf("%d",&n); for(int i=1; i<=n; i++) { scanf("%d",a+i); } build(1,1,n); string cmd; while(cin>>cmd&&cmd!="End") { if(cmd=="Add") { int tmp1,tmp2; scanf("%d%d",&tmp1,&tmp2); update(1,1,n,tmp1,tmp2); } else if(cmd=="Sub") { int tmp1,tmp2; scanf("%d%d",&tmp1,&tmp2); update(1,1,n,tmp1,(-tmp2)); } else if(cmd=="Query") { int tmp1,tmp2; scanf("%d%d",&tmp1,&tmp2); int ans=query(1,1,n,tmp1,tmp2); printf("%d\n",ans); } // print(); } } return 0; }
标签：
原文地址：http://www.cnblogs.com/superxuezhazha/p/5726938.html
