POJ 3468 A Simple Problem with Integers(线段树 ，延迟标记)

题目链接:http://poj.org/problem?id=3468

   今天终于把电脑的输入法修好了!!终于可以多写一点了~具体解释在代码中，本题代码也可以作为线段树区间更新的模板O(∩_∩)O~

#include<cstdio>
#include<algorithm>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define root 1,n,1
#define LL long long//注意本题是long long 型，否则WA
const int maxn=100000+5;
LL add[maxn<<2];//当前区间的子区间需要被加上的值(延迟标记)
LL S[maxn<<2];//当前区间的和
void push_up(int rt)
{
    S[rt]=S[rt<<1]+S[rt<<1|1];//向上传值,累加
}
void push_down(int rt,int dis)
{
    if(add[rt])//如果add做了标记，向下执行
    {
        add[rt<<1]+=add[rt];
        add[rt<<1|1]+=add[rt];//把值赋给左右两棵子树
        S[rt<<1]+=add[rt]*(dis-(dis>>1));
        S[rt<<1|1]+=add[rt]*(dis>>1);//更新区间的值
        add[rt]=0;//删除原来的标记，避免重复执行
    }
}
void build(int l,int r,int rt)//建树
{
    add[rt]=0;//初始化
    if(l==r)
    {
        scanf("%lld",&S[rt]);//将输入的值赋给子节点
        return ;
    }
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    push_up(rt);
}
void update(int L,int R,int d,int l,int r,int rt)//进行区间的值的更新
{
    if(L<=l&&r<=R)//如果要更新的区间包含了[l,r]这个区间
    {
        add[rt]+=d;//将[l,r]区间标记为需要增加的值
        S[rt]+=(LL)(r-l+1)*d;//将值加上去，赋给节点，区间的值更新
        return ;
    }
    push_down(rt,r-l+1);//如果不满足条件，执行push_down函数
    int m=(l+r)>>1;
    if(L<=m)
        update(L,R,d,lson);
    if(m<R)
        update(L,R,d,rson);
    push_up(rt);
}
LL query(int L,int R,int l,int r,int rt)//查询
{
    if(L<=l&&r<=R)//如果要查询的区间包含了[l,r]这个区间
        return S[rt];//直接返回这个区间的值即可
    push_down(rt,r-l+1);//否则继续向子节点传递标记
    int m=(l+r)>>1;
    LL ret=0;
    if(L<=m)//将查询到的所有子区间的值加起来
        ret+=query(L,R,lson);
    if(m<R)
        ret+=query(L,R,rson);
    return ret;//返回，输出
}
int main()
{
    int n,q;
    while(~scanf("%d%d",&n,&q))
    {
        build(root);//首先建立一棵线段树
        while(q--)
        {
            char c[3];
            int x,y,z;
            scanf("%s",c);
            if(c[0]=='Q')
            {
                scanf("%d%d",&x,&y);
                printf("%lld\n",query(x,y,root));
            }
            else
            {
                scanf("%d%d%d",&x,&y,&z);
                update(x,y,z,root);
            }
        }
    }
    return 0;
}