A Simple Problem with Integers POJ - 3468 线段树 区间修改 求和

用到标记数组

#include<cstdio>
#include<cstring>
typedef long long ll;
const int maxn=262144;
int node[maxn/2];  //原数组
ll sum[maxn];     //求和数组
int add[maxn];   //标记数组
void pushup(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void color(int l,int r,int c,int rt)  //标记
{
    sum[rt]+=(ll)c*(r-l+1);
    add[rt]+=c;
}
void push_color(int l,int r,int rt)  //标记下放
{//先判断是否有标记，没有就返回,有的话，就二分区间，给两个子结点染色（打标记），然后把当前标记清空即可。
    if(add[rt])
    {
        int m=(l+r)>>1;
        color(l,m,add[rt],rt<<1);
        color(m+1,r,add[rt],rt<<1|1);
        add[rt]=0;   //取消标记
    }
}
void bulid(int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]=(ll)node[l];
        add[rt]=0;
        return ;
    }
    int m=(l+r)>>1;
    bulid(l,m,rt<<1);
    bulid(m+1,r,rt<<1|1);
    pushup(rt);
}
// 区间增值修改
void update(int L,int R,int c,int l,int r,int rt)  //L~R区间加c
{
/*首先判断修改区间是否完全包含此区间，
包含的话，就直接打标记。
如果不是，那么就下放一次标记，然后再看是否覆盖到了左右子结点。
对对应的子结点修改。
最后再update一下即可。*/
    if(L<=l && R>=r)
    {
        color(l,r,c,rt);
        return ;
    }
    push_color(l,r,rt);
    int m=(l+r)>>1;
    if(L<=m)
        update(L,R,c,l,m,rt<<1);
    if(m<R)
        update(L,R,c,m+1,r,rt<<1|1);
    pushup(rt);
}
ll query(int L,int R,int l,int r,int rt)  L~R区间求和
{
    if(L<=l && r<=R)
        return sum[rt];
    push_color(l,r,rt);
    int m=(l+r)>>1;
    ll ans=0;
    if(L<=m)
        ans+=query(L,R,l,m,rt<<1);
    if(R>m)
        ans+=query(L,R,m+1,r,rt<<1|1);
    return ans;
}
int main()
{
    int n,m;
    char s;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%d",&node[i]);
    bulid(1,n,1);
    int x,y,z;
    for(int i=0;i<m;i++)
    {
        scanf(" %c",&s);
        if(s=='Q')
        {
            scanf("%d%d",&x,&y);
            printf("%lld\n",query(x,y,1,n,1));
        }
        else if(s=='C')
        {
            scanf("%d%d%d",&x,&y,&z);
            update(x,y,z,1,n,1);
        }
    }
    return 0;
}