POJ 3468 A Simple Problem with Integers （線段樹區間更新模板題）

題目鏈接

POJ3468

題目大意

給N個數，Q個操作(1 ≤ N,Q ≤ 100000)
有兩種操作：
1.”C a b c”：Aa , Aa+1 , … , Ab 都加c
2.”Q a b”：求Aa , Aa+1 , … , Ab 的和

分析

線段樹區間更新模板題。

代碼

#include <iostream>
#include <cstring>
#include <string>
#define LL long long
using namespace std;
const int MAXN=100010;
LL a[MAXN],ans[MAXN<<2],lazy[MAXN<<2];

void PushUp(int rt)
{
    ans[rt]=ans[rt<<1]+ans[rt<<1|1];
}
void Build(int l,int r,int rt)
{
    if (l==r)
    {
        ans[rt]=a[l];
        return;
    }
    int mid=(l+r)>>1;
    Build(l,mid,rt<<1);
    Build(mid+1,r,rt<<1|1);
    PushUp(rt);
}
void PushDown(int rt,int ln,int rn)
{
    if (lazy[rt])
    {
        lazy[rt<<1]+=lazy[rt];
        lazy[rt<<1|1]+=lazy[rt];
        ans[rt<<1]+=lazy[rt]*ln;
        ans[rt<<1|1]+=lazy[rt]*rn;
        lazy[rt]=0;
    }
}
void Update(int L,int R,int C,int l,int r,int rt)
{
    if (L<=l&&r<=R)
    {
        ans[rt]+=C*(r-l+1);
        lazy[rt]+=C;
        return;
    }
    int mid=(l+r)>>1;
    PushDown(rt,mid-l+1,r-mid);
    if (L<=mid) Update(L,R,C,l,mid,rt<<1);
    if (R>mid) Update(L,R,C,mid+1,r,rt<<1|1);
    PushUp(rt);
}
LL Query(int L,int R,int l,int r,int rt)
{
    if (L<=l&&r<=R)
        return ans[rt];
    int mid=(l+r)>>1;
    PushDown(rt,mid-l+1,r-mid);
    LL ANS=0;
    if (L<=mid) ANS+=Query(L,R,l,mid,rt<<1);
    if (R>mid) ANS+=Query(L,R,mid+1,r,rt<<1|1);
    return ANS;
}
int main()
{
    ios::sync_with_stdio(false);
    int n,q,i,x,y,z;
    string op;
    cin>>n>>q;
    for (i=1;i<=n;i++)
        cin>>a[i];
    memset(ans,0,sizeof(ans));
    memset(lazy,0,sizeof(lazy));
    Build(1,n,1);
    while (q--)
    {
        cin>>op;
        if (op=="C")
        {
            cin>>x>>y>>z;
            Update(x,y,z,1,n,1);
        }
        if (op=="Q")
        {
            cin>>x>>y;
            cout<<Query(x,y,1,n,1)<<endl;
        }
    }
    return 0;
}