POJ 3468 （线段树）

#include<algorithm>
#include<iostream>
#include<cstdio>
#include<string.h>
using namespace std;
const long long  maxn=100010;
struct node
{
    long long   lazy,sum;
    long long  r,l;
};
node a[maxn<<2];
long long number[maxn];
long long update(long long k)
{
    a[k].sum=a[k<<1].sum+a[k<<1|1].sum;
}
void build(long long  k,long long l,long long  r)
{
    a[k].r=r;
    a[k].l=l;
    a[k].lazy=0;
    if(r==l)
    {
        a[k].sum=number[l];
        return ;
    }
    else
    {
        long long mid=(l+r)>>1;
        build(k<<1,l,mid);
        build(k<<1|1,mid+1,r);
        update(k);
    }
}void down(long long  k)
{
    if(a[k].l==a[k].r)
    {
        a[k].lazy=0;
        return ;
    }
    a[k<<1].lazy+=a[k].lazy;
    a[k<<1|1].lazy+=a[k].lazy;
    a[k<<1].sum+=(a[k<<1].r-a[k<<1].l+1)*a[k].lazy;
    a[k<<1|1].sum+=(a[k<<1|1].r-a[k<<1|1].l+1)*a[k].lazy;
    a[k].lazy=0;
}
void change(long long  k,long long  l,long long   r,long long    y)
{

    long long  mid;
    if(a[k].l==l&&a[k].r==r)
    {
        a[k].lazy+=y;
        a[k].sum+=(r-l+1)*y;
        return ;
    }
    //[1,5]有标记时，修改了 [1,3 ] 的值，如果没有down，在update时 [1,5]的值就错了。
    down(k);
    mid=(a[k].l+a[k].r)>>1;
    if(mid<l)
        change(k<<1|1,l,r,y);
    else if(mid>=r)
        change(k<<1,l,r,y);
    else
    {
        change(k<<1,l,mid,y);
        change(k<<1|1,mid+1,r,y);
    }
    update(k);
}

long long   ans=0;
long long query(long long k,long long  l,long long  r)
{
    if(a[k].l==l&&a[k].r==r)
    {
        return a[k].sum;
    }
    long long  mid=(a[k].r+a[k].l)>>1;
    if(a[k].lazy!=0)
    {
        down(k);
    }

long long res = 0;

    if(mid<l)
      return query(k<<1|1,l,r);
    else if(mid>=r)
       return query(k<<1,l,r);
    else
    {
    return query(k<<1,l,mid)+query(k<<1|1,mid+1,r);
    }


}
int main()
{
    long long  n,m;
    scanf("%lld%lld",&n,&m);
    for(long long i=1; i<=n; i++)
    {
        scanf("%lld",&number[i]);
    }
    build(1,1,n);
    while(m--)
    {
        char h;
        cin>>h;
        long long  q,w;
        long long   e;
        if(h=='Q')
        {
           scanf("%lld%lld",&q,&w);
           ans=query(1,q,w);
            cout<<ans<<endl;
        }
        else
        {

            scanf("%lld%lld%lld",&q,&w,&e);
            change(1,q,w,e);
        }

    }
}