区间修改与区间求和(线段树)

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;
} 
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值