poj 3468 A Simple…

裸的线段树的题目，修改区间，查询区间。

#include <iostream>

#include <stdio.h>

#include <string.h>

#define ls t<<1

#define rs t<<1|1

#define midt (tr[t].l+tr[t].r)>>1

using namespace std;

const int maxn=111111;

long long a[maxn];

struct

{

    int l,r;

    long long sum;

    long long lazy;

}tr[maxn*4];

int maketree(int t,int l,int r)

{

    tr[t].l=l;

    tr[t].r=r;

    tr[t].lazy=0;

    if(l==r)

    {

        tr[t].sum=a[l];

        return(0);

    }

    int mid=midt;

    maketree(ls,l,mid);

    maketree(rs,mid+1,r);

    tr[t].sum=tr[ls].sum+tr[rs].sum;

}

int pushdown(int t)

{

    tr[ls].sum+=tr[t].lazy*(tr[ls].r-tr[ls].l+1);

    tr[ls].lazy+=tr[t].lazy;

    tr[rs].sum+=tr[t].lazy*(tr[rs].r-tr[rs].l+1);

    tr[rs].lazy+=tr[t].lazy;

    tr[t].lazy=0;

}

int pushup(int t,long long tmp)

{

    int tt=(t>>1);

    if(tt==0) return(0);

    tr[tt].sum+=tmp;

    pushup(tt,tmp);

}

int modify(int t,int l,int r,long long tmp)

{

    if(l<=tr[t].l&&r>=tr[t].r)

    {

        tr[t].sum+=tmp*(tr[t].r-tr[t].l+1);

        tr[t].lazy+=tmp;

        pushup(t,tmp*(tr[t].r-tr[t].l+1));

        return(0);

    }

    if(tr[t].lazy)

    pushdown(t);

    int mid=midt;

    if(l<=mid) modify(ls,l,r,tmp);

    if(mid+1<=r) modify(rs,l,r,tmp);

}

long long query(int t,int l,int r)

{

    if(l<=tr[t].l&&r>=tr[t].r)

    return(tr[t].sum);

    if(tr[t].lazy)

    pushdown(t);

    int mid=midt;

    long long ret=0;

    if(l<=mid) ret+=query(ls,l,r);

    if(mid+1<=r) ret+=query(rs,l,r);

    return(ret);

}

int main()

{

    int n,k;

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

    {

        for(int i=1;i<=n;i++)

        scanf("%lld",&a[i]);

        maketree(1,1,n);

        char tmp[10];

        for(int i=1;i<=k;i++)

        {

            int l,r,txt;

            scanf("%s %d %d",tmp,&l,&r);

            if(l>r)

            swap(l,r);

            if(tmp[0]=='Q')

            printf("%lld\n",query(1,l,r));

            else

            {

                scanf("%d",&txt);

                modify(1,l,r,txt);

            }

        }

    }

    return 0;

}