（线段树模板）A Simple Problem with Integers --POJ--3468

链接：

http://poj.org/problem?id=3468

 

代码：

#include<stdio.h>
#include<algorithm>
#include<stdlib.h>
using namespace std;

#define Lson r<<1
#define Rson r<<1|1

const int N = 1e5+5;

struct SegmentTree
{
    int L, R;
    long long sum, e;
    int Mid()
    {
        return (L+R)>>1;
    }
    int len()
    {
        return R-L+1;
    }
}a[N<<2];

void BuildSegTree(int r, int L, int R)
{
    a[r].L=L, a[r].R=R;
    a[r].e=0;

    if(L==R)
    {
        scanf("%lld", &a[r].sum);
        return ;
    }

    BuildSegTree(Lson, L, a[r].Mid());
    BuildSegTree(Rson, a[r].Mid()+1, R);

    a[r].sum = a[Lson].sum + a[Rson].sum;
}
void PushDown(int r)
{
    a[Lson].sum += a[r].e*a[Lson].len();
    a[Lson].e += a[r].e;
    a[Rson].sum += a[r].e*a[Rson].len();
    a[Rson].e += a[r].e;

    a[r].e=0;
}
void Update(int r, int L, int R, int e)
{
    a[r].sum += (R-L+1)*e;

    if(a[r].L==L && a[r].R==R)
    {
        a[r].e += e;
        return ;
    }

    PushDown(r);

    if(R<=a[r].Mid())
        Update(Lson, L, R, e);
    else if(L>a[r].Mid())
       Update(Rson, L, R, e);
    else
    {
        Update(Lson, L, a[r].Mid(), e);
        Update(Rson, a[r].Mid()+1, R, e);
    }
}
long long Query(int r, int L, int R)
{
    if(a[r].L==L && a[r].R==R)
       return a[r].sum;

    PushDown(r);

    if(R<=a[r].Mid())
        return Query(Lson, L, R);
    else if(L > a[r].Mid())
        return Query(Rson, L, R);
    else
    {
        long long Lsum = Query(Lson, L, a[r].Mid());
        long long Rsum = Query(Rson, a[r].Mid()+1, R);

        return Lsum+Rsum;
    }
}

int main()
{
   int n, m;
   while(scanf("%d%d", &n, &m)!=EOF)
   {
       BuildSegTree(1, 1, n);

       while(m--)
       {
           char s[10];
           int L, R, e;

           scanf("%s", s);

           if(s[0]=='Q')
           {
               scanf("%d%d", &L, &R);
               printf("%lld
", Query(1, L, R));
           }
           else
           {
               scanf("%d%d%d", &L, &R, &e);
               Update(1, L, R, e);
           }
       }
   }
   return 0;
}