poj 3468 A Simple Problem with Integers(线段树区间更新）

题意：http://poj.org/problem?id=3468

这纯粹研究区间更新，区间查询，熟练掌握线段树的运用技能。相比，单点更新，多了add标志，adding, query操作时也多了对它的特殊操作pushdown。

#include <iostream>
#include<cstdio>
using namespace std;
#define LL long long
const LL maxn=1e5+5;
struct node{
    LL left,right,sum,add;
}btree[3*maxn];
void build(LL root,LL l,LL r){
    btree[root].left=l;
    btree[root].right=r;
    btree[root].add=0; //add 作为标记，指示哪一个区间的数字需要改变
    if(l==r){
        scanf("%lld",&btree[root].sum);//因为树结构就是从左到右遍历的，所以输入顺序就是完全二叉树叶子节点的左右顺序。
        return ;
    }
    LL mid=(l+r)/2; //注意mid和l,r的关系（不用写成if,else if,else结构） 
    build(2*root,l,mid);
    build(2*root+1,mid+1,r);
    btree[root].sum=btree[2*root].sum+btree[2*root+1].sum;
}
void pushdown(LL root){
    LL dex=root*2;
    btree[dex].add+=btree[root].add;
    btree[dex+1].add+=btree[root].add;
    btree[dex].sum+=btree[root].add*(btree[dex].right-btree[dex].left+1);
    btree[dex+1].sum+=btree[root].add*(btree[dex+1].right-btree[dex+1].left+1);
    btree[root].add=0;
}
void adding(LL root,LL l,LL r,LL val){
    if(r<btree[root].left||l>btree[root].right)return ; //部分越界情况，及时退出
    if(l<=btree[root].left&&r>=btree[root].right){
        btree[root].add+=val;
        btree[root].sum+=val*(btree[root].right-btree[root].left+1);//!! 加上val*(). 
        return ;
    }
    if(btree[root].add)pushdown(root);
    LL mid=(btree[root].left+btree[root].right)/2;
    adding(2*root,l,r,val);  //!! l,r 不是l,mid. 
    adding(2*root+1,l,r,val);
    btree[root].sum=btree[2*root].sum+btree[2*root+1].sum;
}
LL ans=0;
void query(LL root,LL l,LL r){
    if(l>btree[root].right||r<btree[root].left)return ;
    if(l<=btree[root].left&&r>=btree[root].right){
        ans+=btree[root].sum;
        return ;
    }
    if(btree[root].add)pushdown(root);
    LL mid=(btree[root].left+btree[root].right)/2;
    if(r<=mid)query(2*root,l,r);
    else if(l>mid)query(2*root+1,l,r);
    else {
        query(2*root,l,mid);
        query(2*root+1,mid+1,r);
    }
}
int main()
{
    //freopen("cin.txt","r",stdin);
    LL i,n,q;
    while(cin>>n>>q){
        build(1,1,n);
        char s[2];
        while(q--){
            scanf("%s",s);
            if(s[0]=='Q'){
                LL a,b;
                scanf("%lld%lld",&a,&b);
                ans=0;
                query(1,a,b);
                printf("%lld\n",ans);
            }
            else {
                LL a,b,c;
                scanf("%lld%lld%lld",&a,&b,&c);
                adding(1,a,b,c);
            }
        }

    }
    return 0;
}