POJ 3468 线段树裸题

　　这些天一直在看线段树，因为临近期末，所以看得断断续续，弄得有些知识点没能理解得很透切，但我也知道不能钻牛角尖，所以配合着刷题来加深理解。

　　然后，这是线段树裸题，而且是最简单的区间增加与查询，我参考了ACdreamer的模板，在此基础上自己用宏定义来精简了一下代码：

 #include<cstdio>
 typedef long long LL;
 #define root  int rt, int l, int r
 #define lson  rt*2, l, mid
 #define rson  rt*2+1, mid+1, r
 #define makemid  int mid= (l+r)>>1
 int a,b;
 LL c;

 struct tree {
     LL add,sum;
 } s[];

 void build(root) {
     s[rt].add= ;
     if(l==r){
         scanf("%lld",&s[rt].sum);
         return ;
     }
     makemid ;
     build(lson);
     build(rson);
     s[rt].sum= s[rt*].sum+s[rt*+].sum;
 }

 inline void pushdown(int rt, int len) {
     int ls= rt*, rs= ls+;
     const LL &c= s[rt].add;
     s[ls].add += c;
     s[rs].add += c;
     s[ls].sum += c*(len-len/);
     s[rs].sum += c*(len/);
     s[rt].add= ;
 }

 void update(root) {
     if(a<=l && r<=b){
         s[rt].add+= c;
         s[rt].sum+= c*(r-l+);
         return ;
     }
     if(s[rt].add)    pushdown(rt,r-l+);
     makemid ;
     if(a<=mid)    update(lson);
     if(b>mid)    update(rson);
     s[rt].sum= s[rt*].sum+s[rt*+].sum;
 }

 LL query(root) {
     if(a<=l && r<=b)    return s[rt].sum;
     if(s[rt].add)    pushdown(rt,r-l+);
     makemid ;
     LL res= ;
     if(a<=mid)    res+= query(lson);
     if(b>mid)    res+= query(rson);
     return res;
 }

 int main(){
     int n,q;
     while(~scanf("%d%d",&n,&q)){
         build(,,n);
         while(q--){
             getchar();
             if(getchar()=='Q'){
                 scanf("%d%d",&a,&b);
                 printf("%lld\n",query(,,n));
             }
             else {
                 scanf("%d%d%lld",&a,&b,&c);
                 update(,,n);
             }
         }
     }
     return ;
 }

　　原样的代码是：

 #include<cstdio>
 typedef long long LL;
 LL c;
 int a,b;

 struct tree{
     LL add,sum;
 } s[];

 void build(int rt, int l, int r){
     s[rt].add= ;
     if(l==r){
         scanf("%lld",&s[rt].sum);
         return ;
     }
     int mid= (l+r)>>;
     build(rt*,l,mid);
     build(rt*+,mid+,r);
     s[rt].sum= s[rt*].sum+s[rt*+].sum;
 }

 inline void pushdown(int rt, int len){
     if(s[rt].add){
         int ls= rt*, rs= ls+;
         s[ls].add += s[rt].add;
         s[rs].add += s[rt].add;
         s[ls].sum += s[rt].add*(len-len/);
         s[rs].sum += s[rt].add*(len/);
         s[rt].add= ;
     }
 }

 void update(int rt, int l, int r){
     if(a<=l && r<=b){
         s[rt].add+= c;
         s[rt].sum+= c*(r-l+);
         return ;
     }
     pushdown(rt,r-l+);
     int mid= (l+r)>>;
     if(a<=mid)    update(rt*,l,mid);
     if(b>mid)    update(rt*+,mid+,r);
     s[rt].sum= s[rt*].sum+s[rt*+].sum;
 }

 LL query(int rt, int l, int r){
     if(a<=l && r<=b)    return s[rt].sum;
     pushdown(rt,r-l+);
     int mid= (l+r)>>;
     LL res= ;
     if(a<=mid)    res+= query(rt*,l,mid);
     if(b>mid)    res+= query(rt*+,mid+,r);
     return res;
 }

 int main(){
     int n,q;
     while(~scanf("%d%d",&n,&q)){
         build(,,n);
         while(q--){
             getchar();
             if(getchar()=='Q'){
                 scanf("%d%d",&a,&b);
                 printf("%lld\n",query(,,n));
             }
             else {
                 scanf("%d%d%lld",&a,&b,&c);
                 update(,,n);
             }
         }
     }
     return ;
 }

　　线段树，不断进取中~~