一道很经典的线段树题目,lazy-tag标签。在这里大赞notonlysuccess大神的报告,对线段树的描述非常详细,风格也非常的好。
lazy-tag简单的说就是简单来说就是每次更新的时候不要更新到底,用延迟标记使得更新延迟到下次需要更新or询问到的时候。(摘自notonlysuccess)
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#define LS(x) x<<1
#define RS(x) (x<<1)|1
#define Mid(l,r) (l+r)>>1
#define LL long long
const int maxn = 100005;
LL sum[maxn<<2] , add[maxn<<2];
void pushup(int pos)
{
sum[pos] = sum[LS(pos)] + sum[RS(pos)];
}
void pushdown(int pos,int len)
{
if(add[pos])
{
add[LS(pos)] += add[pos];
add[RS(pos)] += add[pos];
sum[LS(pos)] += add[pos] * (len - (len>>1));
sum[RS(pos)] += add[pos] * (len>>1);
add[pos] = 0;
}
}
void build(int l,int r,int pos)
{
add[pos] = 0;
if(l == r)
{
scanf("%lld",&sum[pos]);
return ;
}
int mid = Mid(l,r);
build(l,mid,LS(pos));
build(mid+1,r,RS(pos));
pushup(pos);
}
void Update(int l,int r,int c,int pos,int L,int R)
{
if(l <= L && R <= r)
{
add[pos] += c;
sum[pos] += (LL)c * (R - L + 1);
return ;
}
pushdown(pos,R - L + 1);
int mid = Mid(L,R);
if(mid >= l) Update(l,r,c,LS(pos),L,mid);
if(mid < r) Update(l,r,c,RS(pos),mid+1,R);
pushup(pos);
}
LL Query(int l,int r,int pos,int L,int R)
{
if(l <= L && R <= r)
{
return sum[pos];
}
pushdown(pos,R-L+1);
int mid = Mid(L,R);
LL ans = 0;
if(mid >= l) ans += Query(l,r,LS(pos),L,mid);
if(mid < r) ans += Query(l,r,RS(pos),mid+1,R);
return ans;
}
int main()
{
int N , Q , a, b, c;
scanf("%d%d",&N,&Q);
build(1 , N , 1);
char op[2];
for(int i = 0; i < Q; i ++)
{
scanf("%s",op);
if(op[0] == 'Q'){
scanf("%d %d",&a,&b);
printf("%lld\n",Query(a,b,1,1,N));
} else {
scanf("%d %d %d",&a,&b,&c);
Update(a,b,c,1,1,N);
}
}
return 0;
}