线段树模板

已于 2024-08-02 15:50:22 修改
阅读量258 收藏 1
点赞数 6
分类专栏： 集训 # 2024寒假 文章标签： 算法
于 2024-02-13 18:20:03 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_39530733/article/details/135940572
版权

一、概述

线段树的适用范围很广，可以在线维护修改以及查询区间上的最值，求和。对于线段树来说，每次更新以及查询的时间复杂度为O(logn)。

二、简单线段树（无pushdown）

1.单点修改，区间查询

P3374 【模板】树状数组 1

#include<bits/stdc++.h>
using namespace std;
#define qio ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
typedef long long ll;

const int N = 5e5 + 10;
int a[N];

struct node {
	int l, r;
	ll sum;
} tr[4 * N];

void build(int i, int l, int r) {
	tr[i].l = l, tr[i].r = r;
	if (l == r) {
		tr[i].sum = a[l];
		return;
	}
	int mid = (l + r) >> 1;
	build(i * 2, l, mid);
	build(i * 2 + 1, mid + 1, r);
	tr[i].sum = tr[i * 2].sum + tr[i * 2 + 1].sum;
}

void update(int i, int pos, int value) {
	if (tr[i].l == tr[i].r) {
		tr[i].sum += value;
		return;
	}
	if (pos <= tr[i * 2].r) update(i * 2, pos, value);
	else update(i * 2 + 1, pos, value);
	tr[i].sum = tr[i * 2].sum + tr[i * 2 + 1].sum;
}

ll search(int i, int l, int r) {
	if (l <= tr[i].l && tr[i].r <= r) return tr[i].sum;
	ll res = 0;
	if (tr[i * 2].r >= l) res += search(i * 2, l, r);
	if (tr[i * 2 + 1].l <= r) res += search(i * 2 + 1, l, r);
	return res;
}

int main() {
	int n, m; cin >> n >> m;
	for (int i = 1; i <= n; i++) cin >> a[i];
	build(1, 1, n);
	for (int i = 1; i <= m; i++) {
		int op; cin >> op;
		if (op == 1) {
			int pos, value; cin >> pos >> value;
			update(1, pos, value);
		} else {
			int l, r; cin >> l >> r;
			cout << search(1, l, r) << '\n';
		}
	}
}

2.区间修改，单点查询

P3368 【模板】树状数组 2

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int N = 5e5 + 10;
int a[N];
struct node{
	int l, r;
	int num;
}tr[4 * N];

void build(int i, int l, int r){
	tr[i] = {l, r, 0};
	if(l == r){
		tr[i].num = a[l];
		return;
	}
	int mid = (l + r) >> 1;
	build(i << 1 , l, mid);
	build(i << 1 | 1, mid + 1, r);
}

void update(int i, int l, int r, int x){
	if(l <= tr[i].l && tr[i].r <= r){
		tr[i].num += x;
		return;
	}
	int mid = (tr[i].l + tr[i].r) >> 1;
	if(l <= mid)update(i << 1, l, r, x);
	if(mid < r)update(i << 1 | 1, l, r, x);
}

ll query(int i, int pos){
	if(tr[i].l == tr[i].r) return tr[i].num;
	ll res = 0;
	int mid = (tr[i].l + tr[i].r) >> 1;
	if(pos <= mid) res += query(i << 1, pos);
	else res += query(i << 1 | 1, pos);
	return res + tr[i].num; 
}

int main() {
	int n, m; cin >> n >> m;
	for(int i = 1; i <= n; i++) cin >> a[i];
	build(1, 1, n);
	while(m--){
		int op; cin >> op;
		if(op == 1){
			int l, r, k; cin >> l >> r >> k;
			update(1, l, r, k);
		} else {
			int x; cin >> x;
			cout << query(1, x) << '\n';
		}
	}
}

三、进阶线段树

1.区间修改，区间查询

P3372 【模板】线段树 1 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int N = 1e5 + 10;
int a[N];
struct node{
	int l, r;
	ll num;
	int lz;
}tr[4 * N];

void build(int i, int l, int r){
	tr[i] = {l, r, 0, 0};
	if(l == r){
		tr[i].num = a[l];
		return;
	}
	int mid = (l + r) >> 1;
	build(i << 1 , l, mid);
	build(i << 1 | 1, mid + 1, r);
	tr[i].num = tr[i << 1].num + tr[i << 1 | 1].num;
}

void pushdown(int i){
	if(tr[i].lz){
		tr[i << 1].lz += tr[i].lz;
		tr[i << 1 | 1].lz += tr[i].lz;
		int mid = (tr[i].l + tr[i].r) >> 1;
		tr[i << 1].num += tr[i].lz * (mid - tr[i << 1].l + 1);
		tr[i << 1 | 1].num += tr[i].lz * (tr[i << 1 | 1].r - mid);
		tr[i].lz = 0;
	}
}

void update(int i, int l, int r, int x){
	if(l <= tr[i].l && tr[i].r <= r){
		tr[i].num += x * (tr[i].r - tr[i].l + 1);
		tr[i].lz += x;
		return;
	}
	pushdown(i);
	int mid = (tr[i].l + tr[i].r) >> 1;
	if(l <= mid)update(i << 1, l, r, x);
	if(mid < r)update(i << 1 | 1, l, r, x);
	tr[i].num = tr[i << 1].num + tr[i << 1 | 1].num;
}

ll query(int i, int l, int r){
	if(l <= tr[i].l && tr[i].r <= r) return tr[i].num;
	pushdown(i);
	ll res = 0;
	if(l <= tr[i << 1].r) res += query(i << 1, l, r);
	if(tr[i << 1 | 1].l <= r) res += query(i << 1 | 1, l, r);
	return res;
} 

int main() {
	int n, m; cin >> n >> m;
	for(int i = 1; i <= n; i++) cin >> a[i];
	build(1, 1, n);
	while(m--){
		int op; cin >> op;
		if(op == 1){
			int l, r, k; cin >> l >> r >> k;
			update(1, l, r, k);
		} else {
			int l, r; cin >> l >> r;
			cout << query(1, l, r) << '\n';
		}
	}
}

2.乘法线段树

P3373 【模板】线段树 2 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

先乘后加

#include<bits/stdc++.h>
using namespace std;
#define qio ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
typedef long long ll;

const int N = 1e5 + 10;
int a[N];
ll mod;
struct node{
	int l, r;
	ll num;
	ll plz, mlz;
}tr[4 * N];

void build(int i, int l, int r){
	tr[i] = {l, r, 0, 0, 1};
	if(l == r){
		tr[i].num = a[l] % mod;
		return;
	}
	int mid = (l + r) >> 1;
	build(i << 1 , l, mid);
	build(i << 1 | 1, mid + 1, r);
	tr[i].num = (tr[i << 1].num + tr[i << 1 | 1].num) % mod;
}

void pushdown(int i){
	//下级示数先乘i的mlz标记，再加上i的plz标记
	ll mid = (tr[i].l + tr[i].r) >> 1;
	tr[i << 1].num = (tr[i].mlz * tr[i << 1].num + ((mid - tr[i].l + 1) * tr[i].plz) % mod) % mod;
	tr[i << 1 | 1].num = (tr[i].mlz * tr[i << 1 | 1].num + ((tr[i].r - mid) * tr[i].plz) % mod) % mod;
	//下级mlz标记乘以i的mlz标记
	tr[i << 1].mlz = (tr[i << 1].mlz * tr[i].mlz) % mod;
	tr[i << 1 | 1].mlz = (tr[i << 1 | 1].mlz * tr[i].mlz) % mod;
	//下级plz标记先乘i的mlz标记，再加上i的plz标记
	tr[i << 1].plz = (tr[i << 1].plz * tr[i].mlz + tr[i].plz) % mod;
	tr[i << 1 | 1].plz = (tr[i << 1 | 1].plz * tr[i].mlz + tr[i].plz) % mod;
	//i的mlz，plz标记重置
	tr[i].mlz = 1, tr[i].plz = 0;
}

void add(int i, int l, int r, int x){
	if(l <= tr[i].l && tr[i].r <= r){
		tr[i].num = (tr[i].num + x * (tr[i].r - tr[i].l + 1)) % mod;
		tr[i].plz = (tr[i].plz + x) % mod;
		return;
	}
	pushdown(i);
	int mid = (tr[i].l + tr[i].r) >> 1;
	if(l <= mid) add(i << 1, l, r, x);
	if(mid < r) add(i << 1 | 1, l, r, x);
	tr[i].num = (tr[i << 1].num + tr[i << 1 | 1].num) % mod;
}

void mul(int i, int l, int r, int x){
	if(l <= tr[i].l && tr[i].r <= r){
		tr[i].num = (tr[i].num * x) % mod;
		tr[i].mlz = (tr[i].mlz * x) % mod;
		tr[i].plz = (tr[i].plz * x) % mod;
		return;
	}
	pushdown(i);
	int mid = (tr[i].l + tr[i].r) >> 1;
	if(l <= mid) mul(i << 1, l, r, x);
	if(mid < r) mul(i << 1 | 1, l, r, x);
	tr[i].num = (tr[i << 1].num + tr[i << 1 | 1].num) % mod;
}

ll query(int i, int l, int r){
	if(l <= tr[i].l && tr[i].r <= r) return tr[i].num;
	pushdown(i);
	ll res = 0;
	if(l <= tr[i << 1].r) res = (res + query(i << 1, l, r)) % mod;
	if(tr[i << 1 | 1].l <= r) res = (res + query(i << 1 | 1, l, r)) % mod;
	return res;
} 

int main() {
	qio
	int n, m; cin >> n >> m;
	cin >> mod;
	for(int i = 1; i <= n; i++) cin >> a[i];
	build(1, 1, n);
	while(m--){
		int op; cin >> op;
		if(op == 1){
			int l, r, k; cin >> l >> r >> k;
			mul(1, l, r, k);
		} 
		if(op == 2){
			int l, r, k; cin >> l >> r >> k;
			add(1, l, r, k);
		}
		if(op == 3) {
			int l, r; cin >> l >> r;
			cout << query(1, l, r) << '\n';
		}
	}
}

3.线段树与树链剖分

Problem - 877E - Codeforces

线段树需要在一个序列上进行，最好是一个数组，那如果给定了一棵树该怎么办？

用dfs遍历这棵树，并把遍历的顺序存到dfn中记录搜索的顺序，并记录in数组和out数组表示开始搜索子树i的左边界和右边界，即之后update或者query的l和r。在dfn序列上build线段树即可。

#include<bits/stdc++.h>
using namespace std;
#define qio ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
typedef long long ll;
typedef double db;

const int N = 2e5 + 10;
vector<int> g[N];
int a[N];
struct tree{
	int l, r;
	ll num;
	ll tot;
	int lz;
}tr[N * 4];
//dfn[]表示dfs的顺序，之后将线段树加在dfn[]上
//in[]表示u节点子树在序列上的左边界，out[]表示u节点子树在序列上的有边界
int in[N], out[N], dfn[N], cnt; 

void dfs(int now, int fa){
	//开始遍历now节点子树
	in[now] = ++cnt, dfn[cnt] = now;
	for(auto &nex : g[now]){
		if(nex == fa) continue;
		dfs(nex, now);
	}
	//结束遍历now节点子树
	out[now] = cnt;
}

void build(int i, int l, int r){
	tr[i] = {l, r, 0, 0, 0};
	if(l == r){
		tr[i].num = a[dfn[l]];
		tr[i].tot = 1;
		return;
	}
	int mid = (l + r) >> 1;
	build(i << 1, l, mid);
	build(i << 1 | 1, mid + 1, r);
	tr[i].num = tr[i << 1].num + tr[i << 1 | 1].num;
	tr[i].tot = tr[i << 1].tot + tr[i << 1 | 1].tot;
}
void pushdown(int i){
	if(tr[i].lz){
		//将灯的数量转换
		tr[i << 1].num = tr[i << 1].tot - tr[i << 1].num;
		tr[i << 1 | 1].num = tr[i << 1 | 1].tot - tr[i << 1 | 1].num;
		//lz标记修改
		tr[i << 1].lz ^= 1;
		tr[i << 1 | 1].lz ^= 1;
		tr[i].lz = 0;
	}
}
void update(int i, int l, int r){
	if(l <= tr[i].l && tr[i].r <= r){
		//将灯的数量转换
		tr[i].num = tr[i].tot - tr[i].num;
		//lz标记修改
		//0变1：之前没修改过，现在修改了
		//1变0：之前修改过一次了，再修改一次相当于没修改
		tr[i].lz ^= 1;
		return;
	}
	pushdown(i);
	int mid = (tr[i].l + tr[i].r) >> 1;
	if(l <= mid) update(i << 1, l, r);
	if(mid < r) update(i << 1 | 1, l, r);
	tr[i].num = tr[i << 1].num + tr[i << 1 | 1].num;
	tr[i].tot = tr[i << 1].tot + tr[i << 1 | 1].tot;
}
ll query(int i, int l, int r){
	if(l <= tr[i].l && tr[i].r <= r) return tr[i].num;
	pushdown(i);
	ll res = 0;
	if(l <= tr[i << 1].r) res += query(i << 1, l, r);
	if(tr[i << 1 | 1].l <= r) res += query(i << 1 | 1, l, r);
	return res;
}

signed main() {
	int n; cin >> n;
	for(int i = 2; i <= n; i++){
		int x; cin >> x;
		g[x].push_back(i);
	}
	for(int i = 1; i <= n; i++) cin >> a[i];
	dfs(1, -1);
//	for(int i = 1; i <= cnt; i++){
//		cout << dfn[i] << ' ';
//	}
//	cout << '\n';
//	for(int i = 1; i <= n; i++){
//		cout << in[i] << ' ' << out[i] << '\n';
//	}
	build(1, 1, n);
	int q; cin >> q;
	while(q--){
		string s; cin >> s;
		int x; cin >> x;
		if(s == "pow"){
			update(1, in[x], out[x]);
		}else{
			cout << query(1, in[x], out[x]) << '\n';
		}
	}	
}

四、线段树引申

1.扫描线&&矩形面积井

P5490 【模板】扫描线&&矩形面积并 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

分析详见题解，太难了阿巴阿巴

【学习笔记】扫描线 - 洛谷专栏 (luogu.com.cn)

#include<bits/stdc++.h>
using namespace std;
#define qio ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
typedef long long ll;
#define int ll
const int N = 1e6 + 10;
struct Scanline{
	ll l, r, y;
	//mark标记该线段是入边1还是出边-1
	int mark;
	bool operator < (const Scanline &t) const{
		return y < t.y;
	}
}line[2 * N];
struct Segtree{
	int l, r;
	//num记录该节点，即该划分线段是否还有贡献，num = 0时无贡献
	int num;
	//这个线段区间覆盖的长度
	ll len;
}tr[4 * N];
int X[2 * N];

void build(int i, int l, int r){
	tr[i] = {l, r, 0, 0};
	if(l == r) return;
	int mid = (l + r) >> 1;
	build(i << 1, l, mid);
	build(i << 1 | 1, mid + 1, r);
}
void pushup(int i){
	//如果有贡献
	if(tr[i].num){
		//tr[i].r + 1处理，因为线段树节点为线段，l，r表示的是线段的第1234条
		//所以对于一个叶子节点来说，tr[i].r = tr[i].l，则X也相等
		//而X表示的是坐标，叶子节点线段的真实长度应该为X[tr[i].r + 1] - X[tr[i].l]
		tr[i].len = X[tr[i].r + 1] - X[tr[i].l];
	}else{
		//无贡献则上传左右儿子的结果
		tr[i].len = tr[i << 1].len + tr[i << 1 | 1].len;
	}
}
void update(int i, int l, int r, int mark){
	if(X[tr[i].r + 1] <= l || r <= X[tr[i].l]) return;
	if(l <= X[tr[i].l] && X[tr[i].r + 1] <= r){
		tr[i].num += mark;
		pushup(i);
		return;	
	}
	update(i << 1, l, r, mark);
	update(i << 1 | 1, l, r, mark);
	pushup(i);
}

void solve() {
	int n; cin >> n;
	for(int i = 1; i <= n; i++){
		int x1, y1, x2, y2; cin >> x1 >> y1 >> x2 >> y2;
		//离散化，记录扫描线
		X[(i << 1) - 1] = x1, X[i << 1] = x2;
		line[(i << 1) - 1] = {x1, x2, y1, 1};
		line[i << 1] = {x1, x2, y2, -1};
	}
	n <<= 1;
	//扫描线按y从小到大排序
	sort(line + 1, line + n + 1);
	//X坐标离散化从小到大排序，cnt记录去重后X的个数
	sort(X + 1, X + 1 + n);
	int cnt = unique(X + 1, X + 1 + n) - X - 1;
	//线段的数量一共有cnt - 1条
	build(1, 1, cnt - 1);
	ll ans = 0;
	for(int i = 1; i <= n - 1; i++){
		update(1, line[i].l, line[i].r, line[i].mark);
		ans += tr[1].len * (line[i + 1].y - line[i].y);
	}
	cout << ans << '\n';
}

signed main() {
	qio
	int T = 1;
//	cin >> T;
	while (T--)solve();
}

ov0_
关注
专栏目录
完整的线段树模板，基础模板
04-22
本资源提供了一个完整的线段树基础模板，旨在帮助开发者快速掌握并应用线段树解决实际问题。 特点： 基础性：适合初学者和有一定基础的开发者，从零开始理解线段树的构建和运作原理。 完整性：包含了线段树的构建...
线段树模板题
gulipeng110的博客
03-01 1425
线段树，即把一条线段上存储的数据按照树的形式进行归纳，并且可以进行增删改查的数据结构。线段树的原型还是二叉树，虽然第一次看代码总归会有些手足无措，但是细细品味之后，每一个函数都非常形象，也很有意思。首先，为什么要用线段树：以本题为例，暴力方法直接写的话无非就是录入数组，然后按区间加上k或者求和输出，很明显是O(n^2)的时间，但是如果我们采用树来存储，因为是二叉树，假设有n个数字，那么树深不会超过lg(n)。所以复杂度就是O(nlgn)，对数级的优化，这真是个棒极了的结构，所以我们要学习。
线段树模板（Java）
Easenyang的博客
12-06 2586
  线段树是一种二叉搜索树，与区间树相似，它将一个区间划分成一些单元区间，每个单元区间对应线段树中的一个叶结点。它的主要优势是对于区间求和、区间求最大值、区间修改和单点修改的速度快，时间复杂度能达到O(logN)O(logN)O(logN)。   若以常规的方法在数组中进行区间求和等操作，时间复杂度会达到O(n)O(n)O(n)，若操作的次数量非常大，那么就很容易超时。线段树的优势就体现出来了   线段树的实现基于一维数组，用数组下标 2∗k+12 * k +12∗k+1 的元素代表左儿子，用下标 2∗k+
简单线段树模板
Marcus-Bao的个人主页
09-19 1383
例如:   给你任意几个数,给定N个区间,让你求这个区间的和;简单线段树的运用,帮助我更好的理解线段树,             //线段树基本 #include #define MAXN 100100 #define MINN 10000100 int  num[MAXN],t[MINN]; void build(int L,int R,int d) {      if(L==R)
ZKW线段树模板类
08-06
在ZKW_Segment_Tree.cpp和ZKW_Segment_Tree.h两个文件中，我们可以找到线段树模板类的实现。cpp文件通常包含了函数的具体实现，而h文件则定义了类的接口。类的构造函数可能接收一个整型数组，用于初始化线段树，而...
带懒惰标记的线段树模板
09-02
带懒惰标记的线段树模板
atcoder abc372 启发式合并， dp
2201_75845839的博客
09-22 484
现在考虑如何存数据。可以用并查集合并两个集合，并且在合并的时候一定要将小的集合合并到大的集合中去，因为小集合合并到大集合中，最后的集合一定大于小集合元素数量的两倍，因此合并的时间复杂度不会超过logn。思路：判断当前变化的位置对结果是否会产生影响，在询问前先计算有多少ABC, 每次变换位置x前后扫描一下[x - 2, x + 2]范围内是否有ABC，变换前有让cnt --， 变换后有让cnt ++思路：三进制转换，可以参考二进制，先把当前可以加入的最大的3的幂次加入,这样一定可以凑出答案。
【优选算法】（第一篇）
weixin_73861555的博客
09-25 956
这个⽅法是往后我们学习「快排算法」的时候，「数据划分」过程的重要⼀步。如果将快排算法拆解的话，这⼀段⼩代码就是实现快排算法的「核⼼步骤」。
【算法笔记】二分查找 红蓝染色法
weixin_51325964的博客
09-23 791
一个菜鸟的算法笔记
文档矫正算法：DocTr++
Guo_Python的博客
09-23 382
论文提出了一种新的数据处理方式，解决了以前文档矫正只能处理带有边界信息的完整文档，文章通过数据处理定义了三种类型的训练数据。（a）包含完整文档边界，（b）包含部分文档边界，（c）不包含文档边界。算法框架如下：对于任意弯曲的文档，作者首先通过CNN抽取特征，然后通过transformer结构进行编码和解码，最后用Flow Head预测偏移场，用偏移场矫正弯曲图片。
C语言题目之单身狗2
something for nothing的博客
09-22 498
c语言题目之找单身狗2：在一堆数中找到两个只出现过一次数的数
数位dp(算法篇)
hellmorning的博客
09-23 1019
数位dp也就是将动态规划的思想放入到对数的每一位进行，也是一种动态规划的一种，本篇文章将会带大家简单学习一下
全栈项目小组【算法赛】题目及解题
2201_75648436的博客
09-22 457
全栈项目小组【算法赛】
python 实现linear discriminant analysis线性判别分析算法
最新发布
luthane的博客
09-26 410
线性判别分析（Linear Discriminant Analysis，简称LDA）是一种用于降维和分类的监督学习算法。它通过最大化类间散度和最小化类内散度来找到最佳投影方向，使得不同类别的数据在新的空间中尽可能分开，同一类别的数据尽可能靠近。LDA的基本思想LDA的基本思想是将多维数据映射到低维空间，同时保留数据之间的类别差异。具体来说，LDA通过计算类内散度矩阵（描述同一类别内部数据的分布情况）和类间散度矩阵（描述不同类别之间的差异），然后求解这两个矩阵的广义特征值问题，找到最佳的投影方向。
代码随想录打卡Day39
weixin_52151595的博客
09-21 395
今天是打家劫舍专题，三道题全都看了讲解，第一次做感觉确实是无从下手。。。不过了解了原理之后代码很快就写出来了。
C语言实现线段树模板
本文档提供了一个C语言实现的线段树模板，适合初学者学习和使用。" 线段树是一种二叉树结构，它将一个数组分成多个子区间，并在每个节点上存储这些子区间的某种聚合信息（如和、最大值、最小值等）。这使得我们可以...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值