poj 3468 : A Simple Problem with Integers 【线段树 区间修改】

题目链接

　　题目是对一个数组，支持两种操作

　　　　操作C：对下标从a到b的每个元素，值增加c；

　　　　操作Q：对求下标从a到b的元素值之和。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<functional>
#include<iostream>
#include<cmath>
#include<cctype>
#include<ctime>
using namespace std;
typedef long long LL;

const int maxn=1e5+;
LL lazy[maxn<<],sum[maxn<<];

void push_up(int rt)
{
    sum[rt]=sum[rt<<]+sum[rt<<|];
}
void push_down(int rt,int len)
{
    if(lazy[rt]==) return ;
    lazy[rt<<]+=lazy[rt];
    lazy[rt<<|]+=lazy[rt];
    sum[rt<<]+=lazy[rt]*(len-(len>>));
    sum[rt<<|]+=lazy[rt]*(len>>);
    lazy[rt]=;
}

void build(int rt,int l,int r)
{
    if(l==r)
    {
        lazy[rt]=;
        scanf("%lld",&sum[rt]);
        return ;
    }
    int mid=(l+r)>>;
    build(rt<<,l,mid);
    build(rt<<|,mid+,r);
    push_up(rt);
}

void update(int rt,int l,int r,int ul,int ur,int v)
{
    if(ul<=l&&r<=ur)
    {
        lazy[rt]+=v;
        sum[rt]+=(r-l+)*v;
        return ;
    }
    push_down(rt,r-l+);
    int mid=(l+r)>>;
    if(ul<=mid) update(rt<<,l,mid,ul,ur,v);
    if(ur>mid) update(rt<<|,mid+,r,ul,ur,v);
    push_up(rt);
}

LL query(int rt,int l,int r,int ql,int qr)
{
    if(ql<=l&&r<=qr) return sum[rt];
    int mid=(l+r)>>;
    push_down(rt,r-l+);
    LL ret=;
    if(ql<=mid) ret+=query(rt<<,l,mid,ql,qr);
    if(qr>mid) ret+=query(rt<<|,mid+,r,ql,qr);
    return ret;
}

int main()
{
    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        build(,,n);
        while(m--)
        {
            char op[];
            scanf("%s",op);
            if(op[]=='Q')
            {
                int l,r;
                scanf("%d%d",&l,&r);
                printf("%lld\n",query(,,n,l,r));
            }
            else
            {
                int l,r,c;
                scanf("%d%d%d",&l,&r,&c);
                update(,,n,l,r,c);
            }
        }
    }
}