POJ 3468 A Simple Problem with Integers【線段樹】

#include<iostream>
#include<cstdio>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define mid (l+r)>>1
typedef long long ll;

const int maxn=100000+10;
ll sum[maxn<<2],add[maxn<<2];

inline void read(ll &d){
    char ch;int op=1;
    while(ch=getchar(),(ch>57||ch<48)&&ch!=45);ch==45?op=-1,d=0:d=ch-48;
    while(ch=getchar(),ch>47&&ch<58) d=d*10+ch-48;
    d*=op;
}
inline void write(ll d){
    if(d==0){puts("0");return;}
    if(d<0) putchar(45),d=-d;
    char s[20];s[19]=0;
    int i=18;
    while(d){
        s[i]=d%10+48;
        d/=10;i--;
    }
    puts(s+i+1);
}

inline void pushup(int rt){sum[rt]=sum[rt<<1]+sum[rt<<1|1];}
inline void pushdown(int rt,int m)
{
    if(!add[rt]) return;
    add[rt<<1]+=add[rt];
    add[rt<<1|1]+=add[rt];
    sum[rt<<1]+=add[rt]*(m-(m>>1));
    sum[rt<<1|1]+=add[rt]*(m>>1);
    add[rt]=0;
}
void build(int l,int r,int rt)
{
    add[rt]=0;
    if(l==r){read(sum[rt]);return;}
    int m=mid;
    build(lson);build(rson);
    pushup(rt);
}
void update(int L,int R,ll c,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        add[rt]+=c;
        sum[rt]+=c*(r-l+1);
        return;
    }
    int m=mid;
    pushdown(rt,r-l+1);
    if(L<=m) update(L,R,c,lson);
    if(R>m) update(L,R,c,rson);
    pushup(rt);
}
ll query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R) return sum[rt];
    pushdown(rt,r-l+1);
    int m=mid;
    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;
    scanf("%d%d",&n,&q);
    build(1,n,1);
    char ch;
    for(int i=0;i<q;i++)
    {
        ll x,y,z;
        ch=getchar();
        read(x);
        read(y);
        if(ch=='Q') write(query(x,y,1,n,1));
        else read(z),update(x,y,z,1,n,1);
    }
    return 0;
}