poj 3468 成段加减

poj3468 A Simple Problem with Integers

#include<stdio.h>
#include<string.h>
#define maxn 100010
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define INF 1<<30
__int64 add[maxn<<2];
__int64 sum[maxn<<2];
int n,q;
void pushUp(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void build(int l,int r,int rt)
{
	add[rt]=0;
    if(l==r)
    {
        scanf("%I64d",&sum[rt]);
        return;
    }
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    pushUp(rt);
}
void pushDown(int rt,int m)
{
    if(add[rt])
    {
        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;
    }
}
__int64 query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        return sum[rt];
    }
    pushDown(rt,r-l+1);  //查询时若查询区间与当前区间有交集要先pushDown，不然add[rt]中的值就无法向下更新
    __int64 v=0;
    int m=(l+r)>>1;
    if(L<=m)
        v+=query(L,R,lson);
    if(m<R)
        v+=query(L,R,rson);
    return v;
}
void update(int L,int R,int c,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        add[rt]+=c;
        sum[rt]+=(__int64)(r-l+1)*c;
        return;
    }
    pushDown(rt,r-l+1);//有交集时才向下更新
    int m=(l+r)>>1;
    if(L<=m)
        update(L,R,c,lson);
    if(R>m)
        update(L,R,c,rson);
    pushUp(rt);
}
int main()
{
    int i,j,k,l;
    scanf("%d %d",&n,&q);
    build(1,n,1);
    while(q--)
    {
        char op;
        int a,b,c;
        scanf("%s",&op);
        if(op=='Q')
        {
            scanf("%d %d",&a,&b);
            printf("%I64d\n",query(a,b,1,n,1));
        }
        else
        {
            scanf("%d%d%d",&a,&b,&c);
            update(a,b,c,1,n,1);
        }
    }
    return 0;
}