线段树

于 2024-10-26 11:15:16 发布
阅读量729 收藏 20
点赞数 9
CC 4.0 BY-SA版权
分类专栏： 洛谷 文章标签： 算法 c++ 数据结构
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/jhy20100420zzz/article/details/143250498

文章目录

1 线段树概念

线段树 ( S e g m e n t   T r e e ) (Segment\ Tree) (Segment Tree) 是 O I OI OI 中的常用算法。线段树是一种二叉搜索树，它将一个区间划分成一些单元区间，每个单元区间对应线段树中的一个叶结点。且时间复杂度仅 Θ ( log ⁡ n ) \varTheta(\log n) Θ(logn)。

2 线段树操作

2.1 建树

对于一个区间，我们可以把它用线段树存储。可以把区间 [ 1 , 4 ] [1,4] [1,4] 用如图所示的方式存储：

11 线段树示意

也就是说，如果要求 [ 1 , 4 ] [1,4] [1,4] 的区间和，就可以通过把这个节点的两个儿子 [ 1 , 2 ] [1,2] [1,2] 和 [ 3 , 4 ] [3,4] [3,4] 的权值相加得到。推导到一般式，即

t r e e i . s u m = t r e e i × 2 . s u m + t r e e i × 2 + 1 . s u m tree_i.sum = tree_{i \times 2}.sum + tree_{i \times 2 + 1}.sum treei​.sum=treei×2​.sum+treei×2+1​.sum

struct segment_tree{
    int sum, tag = 0, l, r;
}tree[N << 2];
inline void build(int i, int l, int r) {
    tree[i].l = l, tree[i].r = r;
    if (l == r) {
        tree[i].sum = a[l];
        return;
    } 
    int mid = (l + r) >> 1;
    build(i << 1, l, mid);
    build(i << 1 | 1, mid + 1, r);
    tree[i].sum = tree[i << 1].sum + tree[i << 1 | 1].sum;
}

注意线段树要开四倍空间。

2.2 区间修改

对给定区间每个数增加 k k k。这个似乎有一点棘手，我们可以引入一个新的懒标记 ( L a z y T a g ) (Lazy Tag) (LazyTag)。一个点的懒标记代表它的区间上所有点应当做出的修改。

以 § 1 \texttt§1 §1 中的图为例，例如修改 [ 1 , 3 ] [1,3] [1,3] 的值，从 [ 1 , 4 ] [1,4] [1,4] 向下找，它的左儿子 [ 1 , 2 ] [1,2] [1,2] 显然包含在所要修改的区间之内；它的右儿子 [ 3 , 4 ] [3,4] [3,4] 区间过大，从它的左儿子接着找： [ 3 , 3 ] [3,3] [3,3] 在区间中，而 [ 4 , 4 ] [4,4] [4,4] 不在区间中。所以，只需要把 [ 1.2 ] [1.2] [1.2] 和 [ 3 , 3 ] [3,3] [3,3] 标记。

推导到一般地，如果区间包含在修改区间中，那么它需要打标记；如果不包含，考虑儿子：左儿子在区间中，找左儿子；右儿子在区间中，找右儿子。

如果单纯这样想，显然有一个巨大的纰漏：比如我们修改了 [ 1 , 3 ] [1,3] [1,3]，然后通过查询操作查询了 [ 1 , 3 ] [1,3] [1,3] 的区间和，没有问题；但如果此时查询 [ 2 , 4 ] [2,4] [2,4]，因为 [ 2 , 2 ] [2,2] [2,2] 没有被打标记，所以会少加一个 k k k。

对于一个区间，如果整个区间都要修改，那么首先把点的权值增加 k × ( t r e e i . r − t r e e i . l + 1 ) k \times (tree_i.r - tree_i.l + 1) k×(treei​.r−treei​.l+1)。如果并不是都要修改，这时我们需要一个pushdown操作。所谓pushdown，就是把自己的懒标记清零，再给自己的儿子标记上。

inline void pushdown(int i) {
    if (tree[i].tag) {
        tree[i << 1].tag += tree[i].tag;
        tree[i << 1 | 1].tag += tree[i].tag;
        tree[i << 1].sum += tree[i].tag * (tree[i << 1].r - tree[i << 1].l + 1);
        tree[i << 1 | 1].sum += tree[i].tag * (tree[i << 1 | 1].r - tree[i << 1 | 1].l + 1);
        tree[i].tag = 0;
    }
}

其他的 t a g tag tag 不为 0 0 0 的区间在后面区间查询的时候顺便更新即可。区间修改的代码：

inline void add(int i, int l, int r, int k) {
    if (tree[i].l >= l && tree[i].r <= r) {
        tree[i].sum += k * (tree[i].r - tree[i].l + 1);
        tree[i].tag += k;
        return;
    }
    pushdown(i);
    if (tree[i << 1].r >= l) 
        add(i << 1, l, r, k);
    if (tree[i << 1 | 1].l <= r)
        add(i << 1 | 1, l, r, k);
    tree[i].sum = tree[i << 1].sum + tree[i << 1 | 1].sum;
}

2.3 区间查询

思路类似，注意pushdown即可。

inline int search(int i, int l, int r) {
    if (tree[i].l >= l && tree[i].r <= r) 
        return tree[i].sum;
    pushdown(i);
    int ans = 0;
    if (tree[i << 1].r >= l) 
        ans += search(i << 1, l, r);
    if (tree[i << 1 | 1].l <= r)
        ans += search(i << 1 | 1, l, r);
    return ans;
}

2.4 练习题目

【练习 2.1】P3372 【模板】线段树 1 - 洛谷 | 计算机科学教育新生态

题解

线段树模板题（注意要开 long   long \texttt{long long} long long）。

#include <bits/stdc++.h>
#define endl '\n'
using namespace std;
typedef long long ll;
const ll N = 2e5 + 10;
ll n, m, a[N];
struct segment_tree{
    ll sum, tag = 0, l, r;
}tree[N << 2];
inline void build(ll i, ll l, ll r) {
    tree[i].l = l, tree[i].r = r;
    if (l == r) {
        tree[i].sum = a[l];
        return;
    } 
    ll mid = (l + r) >> 1;
    build(i << 1, l, mid);
    build(i << 1 | 1, mid + 1, r);
    tree[i].sum = tree[i << 1].sum + tree[i << 1 | 1].sum;
}
inline void pushdown(ll i) {
    if (tree[i].tag) {
        tree[i << 1].tag += tree[i].tag;
        tree[i << 1 | 1].tag += tree[i].tag;
        tree[i << 1].sum += tree[i].tag * (tree[i << 1].r - tree[i << 1].l + 1);
        tree[i << 1 | 1].sum += tree[i].tag * (tree[i << 1 | 1].r - tree[i << 1 | 1].l + 1);
        tree[i].tag = 0;
    }
}
inline void add(ll i, ll l, ll r, ll k) {
    if (tree[i].l >= l && tree[i].r <= r) {
        tree[i].sum += k * (tree[i].r - tree[i].l + 1);
        tree[i].tag += k;
        return;
    }
    pushdown(i);
    if (tree[i << 1].r >= l) 
        add(i << 1, l, r, k);
    if (tree[i << 1 | 1].l <= r)
        add(i << 1 | 1, l, r, k);
    tree[i].sum = tree[i << 1].sum + tree[i << 1 | 1].sum;
}
inline ll search(ll i, ll l, ll r) {
    if (tree[i].l >= l && tree[i].r <= r) 
        return tree[i].sum;
    pushdown(i);
    ll ans = 0;
    if (tree[i << 1].r >= l) 
        ans += search(i << 1, l, r);
    if (tree[i << 1 | 1].l <= r)
        ans += search(i << 1 | 1, l, r);
    return ans;
}
int main() {
    cin >> n >> m;
    for (ll i = 1; i<= n; i++) 
        cin >> a[i];
    build(1, 1, n);
    while (m--) {
        ll op, x, y, k;
        cin >> op;
        if (op == 1) {
            cin >> x >> y >> k;
            add(1, x, y, k);
        } else {
            cin >> x >> y;
            cout << search(1, x, y) << endl;
        }
    }
    return 0;
}

3 线段树进阶

3.1 乘法线段树

如果线段树只有乘法操作，那么只要让懒标记变成乘法，然后 tree[i].sum *= k 即可。

但如果是有乘法也有加法，我们需要考虑先加还是先乘。

把懒标记分成加法标记 ( p l z ) (plz) (plz) 和乘法标记 ( m l z ) (mlz) (mlz)。对于 m l z mlz mlz，直接乘上父亲的标记即可，而对于 p l z plz plz，需要把原先的 p l z plz plz 乘上父亲的 m l z mlz mlz 再加上父亲的 p l z plz plz。

【练习 3.1】P3373 【模板】线段树 2 - 洛谷 | 计算机科学教育新生态

题解

乘法线段树模板题（注意要开 long   long \texttt{long long} long long 和取模）。

#include <bits/stdc++.h>
#define endl '\n'
using namespace std;
typedef long long ll;
const ll N = 2e5 + 10;
ll n, m, q, a[N];
struct segment_tree{
    ll sum, plz, mlz, l, r;
}tree[N << 2];
inline ll mod(ll x){
    return (x % m + m) % m;
}
inline void build(ll i, ll l, ll r) {
    tree[i].l = l, tree[i].r = r;
    tree[i].plz = 0, tree[i].mlz = 1;
    if (l == r) {
        tree[i].sum = a[l];
        return;
    } 
    ll mid = (l + r) >> 1;
    build(i << 1, l, mid);
    build(i << 1 | 1, mid + 1, r);
    tree[i].sum = mod(tree[i << 1].sum + tree[i << 1 | 1].sum);
}
inline void pushdown(ll i) {
    if (tree[i].plz == 0 && tree[i].mlz == 1) 
        return;
    int mlz0 = tree[i].mlz, plz0 = tree[i].plz;
    tree[i << 1].sum = mod(tree[i << 1].sum * mlz0 + plz0 * (tree[i << 1].r - tree[i << 1].l + 1));
    tree[i << 1].mlz = mod(tree[i << 1].mlz * mlz0);
    tree[i << 1].plz = mod(tree[i << 1].plz * mlz0 + plz0);
    tree[i << 1 | 1].sum = mod(tree[i << 1 | 1].sum * mlz0 + plz0 * (tree[i << 1 | 1].r - tree[i << 1 | 1].l + 1));
    tree[i << 1 | 1].mlz = mod(tree[i << 1 | 1].mlz * mlz0);
    tree[i << 1 | 1].plz = mod(tree[i << 1 | 1].plz * mlz0 + plz0);
    tree[i].mlz = 1, tree[i].plz = 0;
}
inline void add1(ll i, ll l, ll r, ll k) {
    if (tree[i].l >= l && tree[i].r <= r) {
        tree[i].sum = mod(tree[i].sum * k);
        tree[i].mlz = mod(tree[i].mlz * k);
        tree[i].plz = mod(tree[i].plz * k);
        return;
    }
    pushdown(i);
    if (tree[i << 1].r >= l) 
        add1(i << 1, l, r, k);
    if (tree[i << 1 | 1].l <= r)
        add1(i << 1 | 1, l, r, k);
    tree[i].sum = mod(tree[i << 1].sum + tree[i << 1 | 1].sum);
}
inline void add2(ll i, ll l, ll r, ll k) {
    if (tree[i].l >= l && tree[i].r <= r) {
        tree[i].sum = mod(tree[i].sum + k * (tree[i].r - tree[i].l + 1));
        tree[i].plz = mod(tree[i].plz + k);
        return;
    }
    pushdown(i);
    if (tree[i << 1].r >= l) 
        add2(i << 1, l, r, k);
    if (tree[i << 1 | 1].l <= r)
        add2(i << 1 | 1, l, r, k);
    tree[i].sum = mod(tree[i << 1].sum + tree[i << 1 | 1].sum);
}
inline ll search(ll i, ll l, ll r) {
    if (tree[i].l >= l && tree[i].r <= r) 
        return tree[i].sum;
    pushdown(i);
    ll ans = 0;
    if (tree[i << 1].r >= l) 
        ans = mod(ans + search(i << 1, l, r));
    if (tree[i << 1 | 1].l <= r)
        ans = mod(ans + search(i << 1 | 1, l, r));
    return ans;
}
int main() {
    cin >> n >> q >> m;
    for (ll i = 1; i <= n; i++) 
        cin >> a[i];
    build(1, 1, n);
    while (q--) {
        ll op, x, y, k;
        cin >> op;
        if (op == 1) {
            cin >> x >> y >> k;
            add1(1, x, y, k);
        }
        else if (op == 2) {
            cin >> x >> y >> k;
            add2(1, x, y, k);
        } else {
            cin >> x >> y;
            cout << search(1, x, y) << endl;
        }
    }
    return 0;
}

* 补充：快读快写

模板如下：

template<typename T>inline void read(T &x) {
    bool f = 1;x = 0;char ch = getchar();
    while (ch < '0' || ch > '9') {if (ch == '-') f = !f;ch = getchar(); }
    while (ch >= '0' && ch <= '9') {x = (x << 1) + (x << 3) + (ch ^ 48);ch = getchar();}
    x = (f ? x : -x);return;
}
template<typename T>inline void write(T x) {
    if (x < 0) putchar('-'), x = -x;
    if (x > 9) write(x / 10);
    putchar(x % 10 + '0');return;
}

快写不如printf和优化过的cout快，所以还是建议使用prinf。

4 End

参考文章：

感谢yes_NT和Happy_WA_Day同学的指导。

线段树详解
yjfyjfyjf54188的博客
06-24 1642
因为每一层都要将节点开完，即使它只有一个节点。线段树是一种高效处理区间查询（如区间求和、最小值、最大值等）的数据结构，支持。单点修改虽只修改一个节点，可我们还要修改它的所有祖父。其每一个节点对应原序列的一段区间，它大概是长这样。在使用线段树之前，我们需要预处理每个节点的对应区间，查询区间和单点修改差不多，只是范围变为了一个区间。每个根节点的信息是由左右子节点的信息更新而来。只要是一个区间(包括单点)内的值信息，线段树。个节点，也要多开2倍，所以要开4倍节点。很明显，线段树是一棵树，它是一棵。
【数据结构】 线段树详解
这里是阿安的博客
08-10 1314
本文是个人通过部分博文学习线段树的笔记，并且写一下自己的理解，在文首标明了出处。线段树是把前缀和建在树上，前缀的是子节点的和，修改时只修改与修改点有联系的父节点，这就是线段树在该题的思路，也是线段树产生的背景。
【学习笔记】详解线段树(浅显易懂,匠心之作,图文并茂)
热门推荐
weq2011的博客
01-29 3万+
超详解线段树(浅显易懂,几乎涵盖所有线段树类型讲解,匠心之作,图文并茂)
线段树 (Segment Tree)
ymy的博客
01-04 714
（线段树基本操作）（单点修改 + 区间查询） （区间修改 + 区间查询）（Lazy-Tag）
线段树详解（含代码实现经过测试）
qq_56999918的博客
02-11 2120
1.线段树介绍 什么是线段树？线段树是一种二叉搜索树，与区间树相似，它将一个区间划分成一些单元区间，每个单元区间对应线段树中的一个叶结点。[1] 对于线段树中的每一个非叶子节点[a,b]，它的左儿子表示的区间为[a,(a+b)/2]，右儿子表示的区间为[(a+b)/2+1,b]。因此线段树是平衡二叉树，最后的子节点数目为N，即整个线段区间的长度。 使用线段树可以快速的查找某一个节点在若干条线段中出现的次数，时间复杂度为O(logN）。而未优化的空间复杂度为2N，因此有时需要离散化让空间压缩。--..
线段树建树
12-14
线段树是一种二叉树，也就是说，每个线段都可以用一二叉树表示 比如一个长度为4的线段可以如此表示： ——————————————-4 1————-2————-3————4 1 2 3 4 如果你要表示线段上的和，最上面的根...
第十一届蓝桥杯校内赛：选节目(线段树解法)
01-06
选节目 小明要组织一台晚会，总共准备了 n 个节目。然后晚会的时间有限，他只能最终选择其中的 m 个节目。这 n 个节目是按照小明设想的顺序给定的，顺序不能改变。 小明发现，观众对于晚上的喜欢程度与前几个节目的...
线段树的一种实现
12-25
线段树是一种高效的数据结构，常用于处理区间查询与更新问题。它的核心思想是将一个一维数组通过分治策略转化为一棵二叉树，每个节点代表一个区间的值，这棵树通常具有平衡性质，比如平衡二叉搜索树的特性。线段树在...
线段树模板
02-03
手打了一份线段树代码，用于c++编程， 线段树是一种二叉搜索树，与区间树相似，它将一个区间划分成一些单元区间，每个单元区间对应线段树中的一个叶结点。 使用线段树可以快速的查找某一个节点在若干条线段中出现的...
数据结构：线段树
最新发布
2302_79366101的博客
07-20 906
线段树的节点tree[i]记录了区间i的值,那么可以再定义一个tag[i]，用它统一记录区间i的修改，而不是一个个地修改区间内的每个元素，这个方法称为Lazy-Tag(懒惰标记或延迟标记)。区间修改时给p打了tag标记，接下来就把tag[p]传给左右子树，然后把tag[p]清零。使用Lazy-Tag方法时，若修改的是一个线段区间，就只对这个线段区间进行整体上的。修改，其内部每个元素的内容先不做修改，只有当这个线段区间的一致性被破坏时，才把变。区间i的Lazy操作，用tag[i]记录。
全网最强剖析线段树 | 万字长文爆肝线段树
qq_45832461的博客
03-11 1894
线段树 线段树的基本结构： 线段树中的每个节点都代表一个区间（可以理解为线段），每个节点维护的是父亲的区间二等分后的其中一个子区间 线段树具有唯一的根节点，根节点维护的是整个区间，代表的区间是整个统计范围，比如有序列a[1]~a[N]，那么根节点所代表的区间就是[1,N]。 线段树的每个叶子节点都代表一个长度为1的元区间[x,x] 对于 每个内部节点[l,r]，它的左子节点是[l,mid]，右子节点是[mid+1,r]，其中mid=(l+r)/2向下取整 当有n个元素时，对区间的操作可以在O(logn)O
数据结构与算法（十）线段树(Segment Tree)入门
Chiclaim
06-10 1万+
本文主要包括以下内容： 线段树的概念 线段树的基本操作 实现一个线段树 LeetCode相关线段树的问题 线段树的概念 线段树(Segment Tree)也是一棵树，只不过元素的值代表一个区间。 常用区间的 统计 操作，比如一个区间的最大值(max)，最小值(min)，和(sum)等等 如一个长度为10的数组，它对应的 求和 线段树，如下图所示(图中的数字表示索引)： 根节...
线段树详解(超详细)
Yuanzili2的博客
03-19 439
这篇文章，应该是我写的到目前为止最难也应该是最长的，(也不能说最难吧，反正也不知道怎么说)的一篇文章。其实线段树吧，你之前可能听说过，是一种很难的算法。但是学过之后，又觉得不是那么难(反正我是这样的)。树(数据结构)(必修)位运算(必修)树状数组(选修)好的那么废话不多说我们开始吧。
线段树算法（Segment Tree）
01-01 1232
【代码】线段树基础知识。
线段树（Segment Tree）
m0_50269977的博客
04-27 729
线段树 线段树是用一种树状结构来存储一个连续区间的信息的数据结构。 线段树的几点性质： 1.线段树是平衡（左子树和右子树的高度差最大为1）的二叉树，最大深度为log2n（n为线段树所表示区间的长度） 2.树中的每一个节点代表对应一个区间（叶子节点是一个点…） 3.每个节点（所代表的区间）完全包含它的所有子孙节点 4.对于任意两个节点（所代表的区间）：要么完全包含，要么互不相交 5.在进行区间操作和统计时把区间等价转换成若干个子区间（logn个）的相同操作。 关键在于理解好： 这棵树的每一个结点代表的是一个区
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值