改段求段的模板,首先要理解lazy标记,如果没有它,你每次都需要更新到叶子节点,那时间开销是巨大的。如果加了lazy标记,你每次更新的时候可以将当前节点的孩子所需更新的信息用lazy标记保留,等到下一次需要用到其节点时再去更新。
#include <stdio.h>
#include <ctype.h>
#include <string.h>
#include <stdlib.h>
#include <limits.h>
#include <math.h>
#include <algorithm>
#include <stack>
#include <queue>
#include <vector>
#include <map>
#include <string>
#include <sstream>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
typedef long long LL;
const double pi=4.0*atan(1.0);
const int MAXN=100005;
LL a[MAXN];
LL sum[MAXN<<2];
LL add[MAXN<<2];//lazy标记
void PushUp(int rt)
{
sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void PushDown(int rt,int len)
{
if(add[rt])
{
add[rt<<1]+=add[rt];
add[rt<<1|1]+=add[rt];
sum[rt<<1]+=add[rt]*(len-(len>>1));
sum[rt<<1|1]+=add[rt]*(len>>1);
add[rt]=0;
}
}
void build(int l,int r,int rt)
{
add[rt]=0;
if(l==r)
{
sum[rt]=a[l];
return ;
}
int m=(l+r)>>1;
build(lson);
build(rson);
PushUp(rt);
}
void update(int L,int R,int c,int l,int r,int rt)
{
if(L<=l &&r<=R)
{
add[rt]+=c;
sum[rt]+=(LL)c*(r-l+1);
return ;
}
PushDown(rt,r-l+1);
int m=(l+r)>>1;
//if(R<=m)
// update(L,R,c,lson);
//else if(L>m)
// update(L,R,c,rson);
//else
//{
// update(L,R,c,lson);
// update(L,R,c,rson);
//}
if(L<=m)
update(L,R,c,lson);
if(m<R)
update(L,R,c,rson);
PushUp(rt);
}
LL ans;
void query(int L, int R,int l,int r,int rt)
{
if(L<=l &&r<=R)
{
ans+=sum[rt];
return;
}
PushDown(rt,r-l+1);
int m=(l+r)>>1;
//if(R<=m)
// query(L,R,lson);
//else if(L>m)
// query(L,R,rson);
//else
//{
// query(L,R,lson);
// query(L,R,rson);
//}
if(L<=m)
query(L,R,lson);
if(m<R)
query(L,R,rson);
}
int main()
{
int n,m;
int i,j,k;
int x,y;
char s[50];
while(scanf("%d%d",&n,&m)!=EOF)
{
for(i=1;i<=n;i++)
scanf("%I64d",a+i);
build(1,n,1);
while(m--)
{
scanf("%s%d%d",s,&x,&y);
if(strcmp(s,"Q")==0)
{
ans=0;
query(x,y,1,n,1);
printf("%I64d\n",ans);
}
else
{
scanf("%d",&k);
update(x,y,k,1,n,1);
}
}
}
return 0;
}