poj 3468 线段树求区间和模板

limit 5000MS
时间 2719MS
求区间和模板
两个操作 ： 1. 求区间和
2.区间加法

#include <cstdio>
#include <cstring>
#include <iostream>
#define mem(a) memset(a, 0, sizeof(a))
using namespace std;
const int inf = 0x3f3f3f3f;
const int MAXN =100100;
typedef long long LL;
LL a[MAXN];
LL lazy[MAXN<<2];
struct T
{
    int l, r;
    LL m;
} tree[MAXN<<2];
void pushdown(int k, int d)
{
    if(lazy[k])
    {
        tree[k<<1].m+=lazy[k]*(d - (d>>1));
        tree[k<<1|1].m+=lazy[k]*(d>>1);
        lazy[k<<1]+=lazy[k];
        lazy[k<<1|1]+=lazy[k];
        lazy[k]=0;
    }
    return ;
}
void pushup(int k)
{
    tree[k].m=(tree[k<<1].m+tree[k<<1|1].m);
}
void build(int l, int r, int k)
{
    tree[k].l=l;
    tree[k].r=r;
    tree[k].m=0;
    if(l==r)
    {
        tree[k].m=a[l];
        return;
    }
    int mid = (l+r)>>1;
    build(l, mid, k<<1);
    build(mid+1, r, k<<1|1);
    pushup(k);
}
void update(int l, int r, int k, LL w)
{
    if(tree[k].l>=l&&tree[k].r<=r)
    {
        tree[k].m+=w*(tree[k].r-tree[k].l+1);//直接加上当前区间的长度和
        lazy[k]+=w;
        return;
    }
    pushdown(k,tree[k].r-tree[k].l+1);//先不更新， 到下一次询问或更新时更新，左右子树长度分别为当前长度/2和当前长度减去另一半。
    int mid = (tree[k].l+tree[k].r)>>1;
    if(l<=mid)
        update(l, r, k<<1, w);
    if(r>mid)
        update(l, r, k<<1|1, w);
    pushup(k);
    return ;
}
LL query(int l, int r, int k)
{
    if(tree[k].l>=l&&tree[k].r<=r)
        return tree[k].m;
    pushdown(k, tree[k].r-tree[k].l+1);
    LL ans=0;
    int mid=(tree[k].l+tree[k].r)>>1;
    if(l<=mid)
        ans+=query(l,r,k<<1);
    if(r>mid)
        ans+=query(l,r,k<<1|1);
    pushup(k);
    return ans;
}

int main()
{
    int q, n;
    while(~scanf("%d %d", &n, &q))
    {
        memset(lazy, 0, sizeof(lazy));
        for(int i=1; i<=n; ++i)
            scanf("%lld", &a[i]);
        build(1, n, 1);
        char op[5];
        int l, r;
        LL v;
        while(q--)
        {
            scanf("%s %d %d", op, &l,&r);
            if(op[0]=='Q')
                printf("%lld\n", query(l, r, 1));
            else
            {
                scanf("%lld", &v);
                update(l, r, 1, v);
            }
        }
    }
    return 0;
}