lazy-tag方法。当修改的是一个整块的区间时,只对这个线段区间进行整体上的修改,其内部每个元素的内容先不做修改,只有当这部分线段的一致性被破坏时才能把变化值传递给子区间。那么每次区间修改的复杂度是O(log2n),一共有Q次操作,总复杂度是O(nlog2n)。做lazy操作的子区间,需要记录状态(tag),在下面的代码中使用add[]实现。
#include<stdio.h>
using namespace std;
const int maxn = 1e5 + 10;
using LL = long long;
LL sum[maxn << 2],add[maxn << 2]; //4倍空间
void push_up(int rt){ //向上更新,通过当前节点rt把值递归到父节点
sum[rt] = sum[rt << 1] + sum[rt << 1 | 1];
}
void push_down(int rt,int m){ //更新rt的子节点
if(add[rt]){
add[rt << 1] += add[rt];
add[rt << 1 | 1] += add[rt];
sum[rt << 1] += (m - (m >> 1)) * add[rt];
sum[rt << 1 | 1] += (m >> 1) * add[rt];
add[rt] = 0; //取消本层标记
}
}
#define lson l,mid,rt << 1
#define rson mid + 1,r,rt << 1 | 1
void build(int l, int r, int rt){ //用满二叉树建树
add[rt] = 0;
if(l == r){ //叶子节点,赋值
scanf("lld",&sum[rt]);
return;
}
int mid = (l + r) >> 1;
build(lson);
build(rson);
push_up(rt); //向上更新区间和
}
void update(int a, int b, LL c, int l, int r, int rt){ //区间更新
if(a <= l && b >= r){
sum[rt] += (r - l + 1) * c;
add[rt] += c;
return;
}
push_down(rt,r - l + 1); //向下更新
int mid = (l + r) >> 1;
if(a <= mid) update(a,b,c,lson); //分成两半,继续深入
if(b > mid) update(a,b,c,rson);
push_up(rt); //向上更新
}
LL query(int a, int b, int l, int r, int rt){ //区间求和
if(a <= l && b >= r)return sum[rt]; //满足lazy,直接返回值
push_down(rt,r - l + 1); //向下更新
int mid = (l + r) >> 1;
LL ans = 0;
if(a <= mid)ans += query(a, b, lson);
if(b > mid)ans += query(a, b, rson);
return ans;
}
int main(){
int n,m;
scanf("%d%d",&n,&m);
build(1,n,1);
while(m--){
char str[2];
int a,b;
LL c;
scanf("%s",str);
if(str[0] == 'C'){
scanf("%d%d%lld",&a,&b,&c);
update(a,b,c,1,n,1);
} else {
scanf("%d%d",&a,&b);
printf("%lld\n",query(a,b,1,n,1));
}
}
return 0;
}