POJ 3468 A Simple Problem with Integers

这个题目涉及到线段树的成段更新，中间需要注意变量的类型是 __int64,update函数用来更新[L，R]区间的数，当L<=l&&R>=r时，可以不向下更新，将增加值c存储在记录该节点的对应区间应该增加量的变量中，也就是进行color[rt]+=c操作，等到下次更新该区间时向下更新或者查询区间和时更新。

顺便提醒一下自己：以后遇到用long long ，__int64的题目里面变量最好统一一个类型，不然很容易混淆。

代码：

#include<stdio.h>
#include<string.h>
#define N 100010
#define maxn (1<<20)
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1

typedef __int64  ll;
ll sum[maxn];
ll  ab[N],color[maxn];

void PushUp(ll rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}

void build(ll l,ll r,ll rt)
{
    if(l==r)
    {
        sum[rt]=ab[l];
        return;
    }
    ll m=(l+r)>>1;
    build(lson);
    build(rson);
    PushUp(rt);
}

void PushDown(ll rt,ll m)/*用于向下更新*/
{
    if(color[rt])
    {
        color[rt<<1]+=color[rt];/*对增量进行积累*/
        color[rt<<1|1]+=color[rt];
        sum[rt<<1]+=(m-(m>>1))*color[rt];
        sum[rt<<1|1]+=(m>>1)*color[rt];
        color[rt]=0;
    }
}

void update(ll L,ll R,ll c,ll l,ll r,ll rt)
{
    if(L<=l&&R>=r)
    {
        sum[rt]+=(r-l+1)*c;
        color[rt]+=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);
}

ll query(ll L,ll R,ll l,ll r,ll rt)
{
    if(L<=l&&R>=r)
        return sum[rt];
    ll res=0;
    ll m=(l+r)>>1;
    PushDown(rt,r-l+1);
    if(L<=m)
        res+=query(L,R,lson);
    if(R>m)
        res+=query(L,R,rson);
    return res;
}

int main(void)
{
    ll n,q,i,a,b,c;
    char op[3];
    while(scanf("%lld%lld",&n,&q)!=EOF)
    {
        memset(color,0,sizeof(color));
        for(i=1; i<=n; i++)
            scanf("%lld",ab+i);
        build(1,n,1);
        while(q--)
        {
            scanf("%s%lld%lld",op,&a,&b);
            if(op[0]=='C')
            {
                scanf("%lld",&c);
                update(a,b,c,1,n,1);
            }
            else
            {
                printf("%lld
",query(a,b,1,n,1));
            }
        }
    }
}