HDOJ 1166.敌兵布阵

2015-06-08

问题简述：

　　原题的题意相当于有一些连续摆放的箱子，里面装着球，球的数量可以加减，现要查询几个连续的箱子里球的总数，其中存在放球和拿球的操作。

　　原题链接：http://acm.hdu.edu.cn/showproblem.php?pid=1166

解题思路：

　　查询区间和的问题，可以使用线段树。

　　最初的输入相当于构建线段树的过程；每一次的加减相当于对线段树进行更新；最后使用线段树的查询区间和算法直接得出答案。

源代码：

 /*
 OJ: HDOJ
 ID: forever
 TASK: 1166.敌兵布阵
 LANG: C++
 NOTE: 线段树
 */
 #include <cstdio>

 const int MAX=;
 int a[MAX];
 char str[];

 void build(int l,int r,int flag) {
     if(l==r) {
         scanf("%d",&a[flag]);
         return;
     }
     int m=(l+r)/;
     build(l,m,flag*);
     build(m+,r,flag*+);
     a[flag]=a[flag*]+a[flag*+];
 }

 void update(int x,int y,int l,int r,int flag) {
     if(l==r) {
         a[flag]+=y;
         return;
     }
     int m=(l+r)/;
     if(x<=m)
         update(x,y,l,m,flag*);
     else
         update(x,y,m+,r,flag*+);
     a[flag]=a[flag*]+a[flag*+];
 }

 int query(int x,int y,int l,int r,int flag) {
     if(x<=l&&r<=y)
         return a[flag];
     int m=(l+r)/;
     int ans=;
     if(x<=m)
         ans+=query(x,y,l,m,flag*);
     if(y>m)
         ans+=query(x,y,m+,r,flag*+);
     return ans;
 }

 int main()
 {
     int t,n,x,y,k=;
     scanf("%d",&t);
     while(t--) {
         scanf("%d",&n);
         build(,n,);
         printf("Case %d:\n",k++);
         while(scanf("%s",str)&&str[]!='E') {
             scanf("%d %d",&x,&y);
             if(str[]=='Q')
                 printf("%d\n",query(x,y,,n,));
             else if(str[]=='A')
                 update(x,y,,n,);
             else if(str[]=='S')
                 update(x,-y,,n,);
         }
     }
     return ;
 }