线段树相关知识点总结

丹柒七

已于 2024-08-14 03:26:31 修改

阅读量753

点赞数 7

文章标签：算法数据结构 c语言

于 2024-08-13 23:26:19 首次发布

本文链接：https://blog.csdn.net/m0_75213259/article/details/141173934

版权

引言

线段树是一种树形数据结构，主要应用于处理数组区间操作的问题，比如区间最大致、最小值、和、更新等。线段树适用于需要频繁区间查询和更新的场景。它的时间复杂度为O(logn)，在处理大规模数据时效率非常高。

基本概念

结构

线段树是一棵完全二叉树，用于存储区间信息。树的每个节点代表数组的一个区间（或者说是为子区间），通常使用数组存储。

节点表示

叶子节点表示数组的单个元素。
内部节点表示其子节点的组合信息（如区间和，最大值、最小值）。
树的高度
对于一个长度为n的数组，线段树的高度为O(logn)，因为线段树是一个完全二叉树。

存储方式

通常使用数组来存储线段树。根节点存储在索引1（有事从0开始），对于任一节点i：

左子节点：2 * i或2 * i + 1（如果从0开始）
右子节点：2 * i + 1或2 * i + 2（如果从0开始）
父节点：i / 2 或（i - 1）/ 2（如果从0开始）

线段树的构建

构建线段树的主要步骤如下：

初始化

分配一个足够大的数组来存储线段树，通常大小为4 * n（这是一个经验值，确保能够存储所有节点）。

递归构建

从根节点开始，递归地构建每个节点的信息：

如果是叶子节点，直接存储数组中的值。
如果是内部节点，计算左子树和右子树的组合信息。

void build(int l, int r, int rt, int heights[], int n, int* tree) {
    if (l == r) {
        tree[rt] = heights[l - 1]; // 叶子节点
        return;
    }
    int mid = (l + r) / 2;
    build(l, mid, rt * 2, heights, n, tree); // 递归构建左子树
    build(mid + 1, r, rt * 2 + 1, heights, n, tree); // 递归构建右子树
    tree[rt] = max(tree[rt * 2], tree[rt * 2 + 1]); // 存储左右子树的最大值
}

区间查询

区间查询用于在数组的某个区间内找到特定的信息，比如最大值，最小值或区间和。

基本思想

通过递归的方式，判断当前节点的区间是否与查询区间有交集。
如果完全子啊查询区间内，直接返回该节点存储的值。
如果没有交集，返回一个无效值（例如0或-∞）
如果部分重叠，递归查询左子树和右子树，并合并结果。

int query(int L, int R, int l, int r, int rt, int* tree) {
    if (L <= l && r <= R) { // 当前区间完全在查询区间内
        return tree[rt];
    }
    int mid = (l + r) / 2;
    int res = 0; // 初始化查询结果（根据问题需要调整）
    if (L <= mid) { // 查询左子树
        res = max(res, query(L, R, l, mid, rt * 2, tree));
    }
    if (mid < R) { // 查询右子树
        res = max(res, query(L, R, mid + 1, r, rt * 2 + 1, tree));
    }
    return res;
}

更新操作

当数组中的某个元素发生变化时，需要更新线段树以反映这个变化。

更新思想

从根节点开始递归，找到需要跟新的叶子节点。
更新该叶子节点的值，然后向上更新父节点的值，直到根节点。

void update(int pos, int val, int l, int r, int rt, int* tree) {
    if (l == r) {
        tree[rt] = val; // 更新叶子节点
        return;
    }
    int mid = (l + r) / 2;
    if (pos <= mid) { // 更新左子树
        update(pos, val, l, mid, rt * 2, tree);
    } else { // 更新右子树
        update(pos, val, mid + 1, r, rt * 2 + 1, tree);
    }
    tree[rt] = max(tree[rt * 2], tree[rt * 2 + 1]); // 更新父节点
}