poj_3468,线段树成段更新

#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;
}