CODEVS 1081 线段树区间修改

本文详细介绍了递归线段树的实现模板,包括区间修改和查询操作,适合算法初学者理解和掌握。

一份经典的递归线段树模版

[简单区间修改]

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
int f[400010],a[100010],n,x,q;
void change(int now,int l,int r,int l1,int r1,int va){
	if (l==l1&&r==r1) f[now]+=va;
	else {
		f[now*2]+=f[now];f[now*2+1]+=f[now];
		f[now]=0;
		int mid=(l+r)/2;
		if (mid>=r1) change(now*2,l,mid,l1,r1,va);
		else if (mid<l1) change(now*2+1,mid+1,r,l1,r1,va);
		else {
			change(now*2,l,mid,l1,mid,va);
			change(now*2+1,mid+1,r,mid+1,r1,va);
		}
	}
}
int query(int now,int l,int r,int tar){
	if (l==r) return f[now];
	else {
		int mid=(l+r)/2;
		if (mid>=tar) return query(now*2,l,mid,tar)+f[now];
		else return query(now*2+1,mid+1,r,tar)+f[now];
	}
}
int main(){
	memset(f,0,sizeof(f));
	scanf("%d",&n);
	for (int i=1;i<=n;i++) scanf("%d",&a[i]);
	scanf("%d",&q);
	for (int i=1;i<=q;i++) {
		int l,r,v,y;
		scanf("%d",&x);
		if (x==1){
			scanf("%d%d%d",&l,&r,&v);
			change(1,1,n,l,r,v);
		}
		if (x==2){
			scanf("%d",&y);
			printf("%d\n",query(1,1,n,y)+a[y]);
		}
	}
	return 0;
}


### 线段树区间修改的实现方法 线段树是一种高效的据结构,用于处理动态区间查询和修改操作。对于区间修改操作,通常会引入懒惰传播(Lazy Propagation)机制来优化性能。 #### 基本概念 在支持区间修改的情况下,线段树通过维护一个额外的`lazy[]`组记录尚未传递给子节点的延迟更新信息。这种设计可以减少不必要的递归调用次,从而提高效率[^1]。 #### 关键函说明 以下是实现线段树区间修改的核心部分: 1. **构建线段树** 构建过程与普通的线段树相同,初始化时需确保`lazy[]`组全部置零。 2. **Push Down 函** 当访问某个节点并发现该节点存在未解决的延迟标记时,需要将其影响向下传递至子节点,并清除当前节点上的标记。 ```cpp void pushDown(int k, int l, int r) { if (lazy[k]) { // 如果有延迟标记 int mid = (l + r) / 2; add(k * 2, l, mid, lazy[k]); // 更新左孩子 add(k * 2 + 1, mid + 1, r, lazy[k]); // 更新右孩子 lazy[k] = 0; // 清除当前节点的延迟标记 } } ``` 3. **Add 函** `add()`负责执行具体的区间修改逻辑。如果目标区间完全覆盖当前节点,则直接应用修改;否则继续分解到子节点上。 ```cpp void add(int k, int l, int r, int ql, int qr, int val) { if (ql <= l && r <= qr) { // 完全覆盖的情况 s[k] += (r - l + 1) * val; // 修改当前区间的总和 lazy[k] += val; // 设置延迟标记 return; } int mid = (l + r) / 2; pushDown(k, l, r); // 下推延迟标记 if (ql <= mid) add(k * 2, l, mid, ql, qr, val); if (qr > mid) add(k * 2 + 1, mid + 1, r, ql, qr, val); s[k] = s[k * 2] + s[k * 2 + 1]; // 合并左右孩子的结果 } ``` 4. **Query 函** 查询过程中也需要注意是否存在延迟标记,若有则先进行下推操作再继续查找。 ```cpp long long query(int k, int l, int r, int ql, int qr) { if (ql <= l && r <= qr) { // 完全覆盖的情况 return s[k]; } int mid = (l + r) / 2; pushDown(k, l, r); // 下推延迟标记 long long res = 0; if (ql <= mid) res += query(k * 2, l, mid, ql, qr); if (qr > mid) res += query(k * 2 + 1, mid + 1, r, ql, qr); return res; } ``` 以上即为完整的线段树区间修改实现方案[^2][^3]。 ### 时间复杂度分析 每次修改或查询操作最多涉及从根节点到叶子节点的一条路径上的所有节点,因此时间复杂度均为\( O(\log N) \)[^4]。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值