hdu1166 敌兵布阵 树状数组/线段树

数列的单点修改、区间求和

树状数组或线段树入门题

 #include<stdio.h>
 #include<string.h>

 int c[],N;

 void add(int x,int a){
     while(x<=N){
         c[x]+=a;
         x+=(x&-x);
     }
     return;
 }

 int sum(int x){
     int t=;
     while(x){
         t+=c[x];
         x-=(x&-x);
     }
     return t;
 }

 int main(){
     int T;
     while(scanf("%d",&T)!=EOF){
         for(int q=;q<=T;q++){
             printf("Case %d:\n",q);
             memset(c,,sizeof(c));
             scanf("%d",&N);
             int i,t;
             for(i=;i<=N;i++){
                 scanf("%d",&t);
                 add(i,t);
             }
             char s[];
             while(scanf("%s",s)){
             //    printf("%s\n",s);
                 if(s[]=='S'){
                     scanf("%d%d",&i,&t);
                     add(i,-t);
                 }
                 else if(s[]=='A'){
                     scanf("%d%d",&i,&t);
                     add(i,t);
                 }
                 else if(s[]=='Q'){
                     scanf("%d%d",&i,&t);
                     printf("%d\n",sum(t)-sum(i-));
                 }
                 else if(s[]=='E')break;
             }
         }
     }
     return ;
 }

树状数组

 #include<stdio.h>
 #include<string.h>
 const int maxm=;

 char s[];
 int a[maxm],st[maxm<<];

 void build(int o,int l,int r){
     if(l==r){
         st[o]=a[l];
         return;
     }
     int m=l+((r-l)>>);
     build(o<<,l,m);
     build(o<<|,m+,r);
     st[o]=st[o<<]+st[o<<|];
 }

 void update(int o,int l,int r,int x,int c){
     if(l==r){
         st[o]+=c;
         return;
     }
     int m=l+((r-l)>>);
     if(x<=m)update(o<<,l,m,x,c);
     if(x>=m+)update(o<<|,m+,r,x,c);
     st[o]=st[o<<]+st[o<<|];
 }

 int query(int o,int l,int r,int ql,int qr){
     if(ql<=l&&qr>=r)return st[o];
     int m=l+((r-l)>>);
     int ans=;
     if(ql<=m)ans+=query(o<<,l,m,ql,qr);
     if(qr>=m+)ans+=query(o<<|,m+,r,ql,qr);
     return ans;
 }

 int read(){
     int x=;
     char c=getchar();
     while(c>''||c<'')c=getchar();
     while(c>=''&&c<=''){
         x=x*+c-'';
         c=getchar();
     }
     return x;
 }

 int main(){
     int T=read();
     for(int q=;q<=T;q++){
         int n=read();
         int i;
         for(i=;i<=n;i++)a[i]=read();
         build(,,n);
         printf("Case %d:\n",q);
         while(){
             scanf("%s",s);
             if(s[]=='Q'){
                 int ql=read();
                 int qr=read();
                 printf("%d\n",query(,,n,ql,qr));
             }
             else if(s[]=='A'){
                 int x=read();
                 int c=read();
                 update(,,n,x,c);
             }
             else if(s[]=='S'){
                 int x=read();
                 int c=read();
                 update(,,n,x,-c);
             }
             else if(s[]=='E')break;
         }
     }
     return ;
 }

线段树