poj_3468,线段树成段更新

参考自http://www.notonlysuccess.com/index.php/segment-tree-complete/
#include<iostream>

#include<cstdio>

#include<cstring>

#include<algorithm>

#define lson l,m,rt<<1

#define rson m+1,r,rt<<1|1

using namespace std;

const int maxn=111111;

long long  sum[maxn<<2];

long long  col[maxn<<2];

void pushUp(int rt)

{

    sum[rt]=sum[rt<<1]+sum[rt<<1|1];

}

void pushDown(int rt,int m)

{

    if(col[rt])

    {

        col[rt<<1]+=col[rt];

        col[rt<<1|1]+=col[rt];

        sum[rt<<1]+=col[rt]*(m-(m>>1));

        sum[rt<<1|1]+=col[rt]*(m>>1);

        col[rt]=0;

    }

}

void build(int l,int r,int rt)

{

    col[rt]=0;

    if(r==l)

    {

        scanf("%lld",&sum[rt]);

        return ;

    }

    int m=(r+l)>>1;

    build(lson);

    build(rson);

    pushUp(rt);

}

void update(int L,int R,int d,int l,int r,int rt)

{

    if(L<=l&&r<=R)

    {

        sum[rt]+=d*(r-l+1);

        col[rt]+=d;

        return;

    }

    pushDown(rt,r-l+1);

    int m=(l+r)>>1;

    if(L<=m) update(L,R,d,lson);

    if(m<R) update(L,R,d,rson);

    pushUp(rt);

}

long long query(int L,int R,int l,int r,int rt)

{

    long long ret=0;

    if(L<=l&&r<=R)

        {

        return sum[rt];

        }

    pushDown(rt,r-l+1);

    int m=(r+l)>>1;

    if(L<=m) ret+=query(L,R,lson);

    if(m<R) ret+=query(L,R,rson);

    return ret;

}

int main()

{

    int n,m;

    while(scanf("%d%d",&n,&m)!=EOF)

    {

        build(1,n,1);

        char c;

        long long  a,b,d;

        for(int i=0;i<m;i++)

        {

            getchar();

            scanf("%c%lld%lld",&c,&a,&b);

            if(c=='Q')

                printf("%lld\n",query(a,b,1,n,1));

            else

            {

                scanf("%lld",&d);

                update(a,b,d,1,n,1);

            }

        }

    }

    return 0;

}
巴特西

poj_3468,线段树成段更新

最新文章

热门文章
