
线段树复习笔记
Published on | 7/29/2025 |
Created on | 10/21/2021 |
By Chesium 2021/10/21
-
线段树能够维护满足以下条件的区间问题:
即区间 的答案可以通过合并 和 的答案得到。 如这些可以:区间和、区间最大值。 如这些不行:区间众数、区间最长不下降子序列
-
数组 长度开至 即可。
-
建树:递归思想
- 时间复杂度:
- 参数:对应数组闭区间 ,以及当前处理的 数组索引 ( 一开始是 ,不能为 )
- 若 ,就不用再细分了,直接令 ,返回。
- 否则,则二分处理:令 ,即为数组中间索引,将 分成 和 两部分。分别建树,两者对应的索引 分别为 和
- 建完树后,我们便把上述两区间合并,形成当前正在建立的区间的值。
-
单点修改:DFS
- 时间复杂度:
- 参数:需更新元素索引 ,修改值 ,当前节点对应区间 ,当前节点编号( 数组索引) 。
- 若 ,则当前区间中只有一个元素,肯定是我们要改的,直接修改 ,返回。
- 计算出中位索引 。
- 若 ,则说明修改值在 中,否则在 中, 递归更新。
- 回溯时更新自己的值 。
-
无更新查询:递归思想
- 时间复杂度:
- 参数:查询区间 ,当前节点对应区间 ,当前节点编号( 数组索引) 。
- 可以理解为求查询区间与当前节点区间交集的对应值
- 若 ,即 ,当前区间必是最终查询值的组成部分,称为查询区间的一个极大区间,直接返回自己的值 。
- 否则,二分为上面讲过的两区间 和 ,不改变 ,进行查询。
- 若 ,则 ,需要考虑,查询 。
- 若 ,则 ,需要考虑,查询 。
- 回溯时合并区间即可。
-
区间修改:懒惰标记
- 时间复杂度:
- 目的:通过延迟对节点信息的更改,从而减少可能不必要的操作次数。
- 参数:修改区间 ,修改值 ,当前节点对应区间 ,当前节点编号( 数组索引) 。
- 若 ,即 ,当前区间是修改区间的子集,肯定要改:修改 ,添加标记 。(更新标记,某些操作为往上叠加(如增加),有些是覆盖(如赋值)),返回。
- 若有标记存在则进行标记下传(防止父区间的懒惰标记下传时覆盖子区间的懒惰标记)
- 类似上述的无更新查询,分别考虑二分后的两区间是否需要进行操作。
- 回溯时记得合并区间!
-
标记下传
- 时间复杂度:
- 参数:当前节点对应区间 ,当前节点编号( 数组索引) 。
- 若无标记,则直接返回。
- 若 ,则标记无作用,也没有区间用于下传,直接返回。
- 求出 ,根据区间范围修改 与 的值
- 更新子区间的懒惰标记 和
- 清除自身的懒惰标记。
-
带修改区间查询
- 时间复杂度: 基本同无修改版本,只不过在二分求解前进行一次标记下传。
-
参考代码
#include <cctype>
#include <cstdio>
#include <cstring>
#include <iostream>
//---//
#include <algorithm>
#include <cmath>
#include <vector>
using namespace std;
typedef unsigned int u;
typedef long long ll;
typedef unsigned long long llu;
#define rep(i, a, b) for (ll i = a; i < b; i++)
#define REP(i, a, b) for (ll i = a; i <= b; i++)
#define per(i, b, a) for (ll i = b; i >= a; i--)
// P3372 【模板】线段树 1
#define get_mid ll m = s + ((t - s) >> 1)
const ll N = 1e5 + 10;
ll a[N], d[4 * N], b[4 * N];
bool lz[4 * N];
void build(ll s, ll t, ll p) {
if (s == t) {
d[p] = a[s];
return;
}
get_mid;
build(s, m, 2 * p);
build(m + 1, t, 2 * p + 1);
d[p] = d[2 * p] + d[2 * p + 1];
}
ll push_down(ll s, ll t, ll p) {
get_mid;
if (s == t || (!lz[p])) return m;
b[2 * p] += b[p];
b[2 * p + 1] += b[p];
d[2 * p] += b[p] * (m - s + 1);
d[2 * p + 1] += b[p] * (t - m);
lz[2 * p] = true;
lz[2 * p + 1] = true;
b[p] = 0;
lz[p] = false;
return m;
}
void update(ll l, ll r, ll c, ll s, ll t, ll p) {
if (l <= s && t <= r) {
d[p] += (t - s + 1) * c;
b[p] += c;
lz[p] = true;
return;
}
ll m = push_down(s, t, p);
if (l <= m) update(l, r, c, s, m, 2 * p);
if (r > m) update(l, r, c, m + 1, t, 2 * p + 1);
d[p] = d[2 * p] + d[2 * p + 1];
}
ll query(ll l, ll r, ll s, ll t, ll p) {
if (l <= s && t <= r) {
return d[p];
}
ll sum = 0, m = push_down(s, t, p);
if (l <= m) sum += query(l, r, s, m, 2 * p);
if (r > m) sum += query(l, r, m + 1, t, 2 * p + 1);
return sum;
}
signed main() {
ll n, m, t, l, r, k;
scanf("%lld%lld", &n, &m);
REP(i, 1, n) scanf("%lld", &a[i]);
build(1, n, 1);
while (m--) {
scanf("%lld%lld%lld", &t, &l, &r);
if (t == 1) {
scanf("%lld", &k);
update(l, r, k, 1, n, 1);
} else
printf("%lld\n", query(l, r, 1, n, 1));
}
}
// https://www.luogu.com.cn/record/60473191
- P3373 【模板】线段树 2 - 洛谷这道题中有两种区间修改操作:加法和乘法,我们需要两个懒惰标记来分别记录它们。 标记下传中应该先乘后加: 如将序列先 ,再 ,再 ,则新的元素和应为: 由于 ,所以将序列 时应将 值也 。
- 参考代码:
#include <cctype>
#include <cstdio>
#include <cstring>
#include <iostream>
//---//
#include <algorithm>
#include <cmath>
#include <vector>
using namespace std;
typedef unsigned int u;
typedef long long ll;
typedef unsigned long long llu;
#define rep(i, a, b) for (ll i = a; i < b; i++)
#define REP(i, a, b) for (ll i = a; i <= b; i++)
#define per(i, b, a) for (ll i = b; i >= a; i--)
// P3372 【模板】线段树 2
#define get_mid ll m = s + ((t - s) >> 1)
const ll N = 1e5 + 10;
ll mod, a[N], d[4 * N], add[4 * N], mul[4 * N];
bool lz[4 * N];
void build(ll s, ll t, ll p) {
mul[p] = 1;
if (s == t) {
d[p] = a[s];
return;
}
get_mid;
build(s, m, 2 * p);
build(m + 1, t, 2 * p + 1);
d[p] = (d[2 * p] + d[2 * p + 1]) % mod;
}
ll push_down(ll s, ll t, ll p) {
get_mid;
if (s == t || (!lz[p])) return m;
d[2 * p] = (add[p] * (m - s + 1) % mod + d[2 * p] * mul[p] % mod) % mod;
d[2 * p + 1] = (add[p] * (t - m) % mod + d[2 * p + 1] * mul[p] % mod) % mod;
add[2 * p] = (add[2 * p] * mul[p] % mod + add[p]) % mod;
add[2 * p + 1] = (add[2 * p + 1] * mul[p] % mod + add[p]) % mod;
mul[2 * p] = mul[2 * p] * mul[p] % mod;
mul[2 * p + 1] = mul[2 * p + 1] * mul[p] % mod;
lz[2 * p] = true;
lz[2 * p + 1] = true;
add[p] = 0;
mul[p] = 1;
lz[p] = false;
return m;
}
void update_add(ll l, ll r, ll c, ll s, ll t, ll p) {
if (l <= s && t <= r) {
d[p] = ((t - s + 1) * c % mod + d[p]) % mod;
add[p] = (c + add[p]) % mod;
lz[p] = true;
return;
}
ll m = push_down(s, t, p);
if (l <= m) update_add(l, r, c, s, m, 2 * p);
if (r > m) update_add(l, r, c, m + 1, t, 2 * p + 1);
d[p] = (d[2 * p] + d[2 * p + 1]) % mod;
}
void update_mul(ll l, ll r, ll c, ll s, ll t, ll p) {
if (l <= s && t <= r) {
d[p] = d[p] * c % mod;
add[p] = add[p] * c % mod;
mul[p] = (mul[p] * c) % mod;
lz[p] = true;
return;
}
ll m = push_down(s, t, p);
if (l <= m) update_mul(l, r, c, s, m, 2 * p);
if (r > m) update_mul(l, r, c, m + 1, t, 2 * p + 1);
d[p] = (d[2 * p] + d[2 * p + 1]) % mod;
}
ll query(ll l, ll r, ll s, ll t, ll p) {
if (l <= s && t <= r) return d[p];
ll sum = 0, m = push_down(s, t, p);
if (l <= m) sum = (sum + query(l, r, s, m, 2 * p)) % mod;
if (r > m) sum = (sum + query(l, r, m + 1, t, 2 * p + 1)) % mod;
return sum;
}
signed main() {
ll n, m, t, l, r, k;
scanf("%lld%lld%lld", &n, &m, &mod);
REP(i, 1, n) scanf("%lld", &a[i]);
build(1, n, 1);
while (m--) {
scanf("%lld%lld%lld", &t, &l, &r);
switch (t) {
case 1:
scanf("%lld", &k);
update_mul(l, r, k, 1, n, 1);
break;
case 2:
scanf("%lld", &k);
update_add(l, r, k, 1, n, 1);
break;
case 3:
printf("%lld\n", query(l, r, 1, n, 1));
break;
}
}
}
// https://www.luogu.com.cn/record/60476253
- 参考