HDU 1166 - 敌兵布阵 - [单点修改、区间查询zkw线段树]

题还是那个题：http://www.cnblogs.com/dilthey/p/6827959.html

不过我们今天换一种线段树实现来做这道题；

关于zkw线段树的讲解：https://zhuanlan.zhihu.com/p/29876526（而且我还在文章里被@了，超开心的ヾ(≧∇≦*)ヾ）

可以说是讲解非常的清楚了！

AC代码（包含了拥有节点更新、区间查询功能的zkw线段树模板）：

 #include<cstdio>
 #include<cstring>
 #define MAX 131072
 int n,m;
 int tree[MAX];
 void pushup(int rt){tree[rt]=tree[rt<<]+tree[rt<<|];}
 void build()
 {
     for(m=;m<n;m<<=);
     for(int i=m;i<=*m-;i++) tree[i]=;//初始化叶子结点
     for(int i=m+;i<=m+n;i++) scanf("%d",&tree[i]);//存储单点值到叶子结点
     for(int i=m-;i;i--) pushup(i);
 }
 void update(int pos,int val)
 {
     pos+=m;
     tree[pos]+=val;//更新叶子结点
     for(pos>>=;pos;pos>>=) pushup(pos);
 }
 int query(int l,int r)
 {
     int ret=;
     for(l=l+m-,r=r+m+;l^r^;l>>=,r>>=)
     {
         if(~l&) ret+=tree[l^];
         if(r&) ret+=tree[r^];
     }
     return ret;
 }
 int main()
 {
     int t;
     scanf("%d",&t);
     for(int kase=;kase<=t;kase++)
     {
         scanf("%d",&n);
         build();
         printf("Case %d:\n",kase);
         char input[];
         while()
         {
             scanf("%s",input);
             if(input[]=='E') break;
             if(input[]=='Q')
             {
                 int l,r;
                 scanf("%d%d",&l,&r);
                 printf("%d\n",query(l,r));
             }
             if(input[]=='A')
             {
                 int pos,x;
                 scanf("%d%d",&pos,&x);
                 update(pos,x);
             }
             if(input[]=='S')
             {
                 int pos,x;
                 scanf("%d%d",&pos,&x);
                 update(pos,-x);
             }
         }
     }
 }

PS. MAX=131072这个也是比较有讲究的，因为n最大是50000，所以对应的我们的m算出来是65536，所以这棵完全二叉树最后一个叶子节点的编号为2 * 65536 - 1 = 131072 - 1.

PS2.其实不难发现，zkw线段树也是一种用空间换取时间和代码量的做法.