c++中如何实现数组的区间求和查询_c++树状数组与线段树简介【详解】

9次阅读

因为for循环区间求和O(n)无法应对高频查询或修改,树状数组(单点改+区间和)和线段树(支持区间更新、最值等复杂操作)通过预处理将查询/更新优化至O(log n)。

c++中如何实现数组的区间求和查询_c++树状数组与线段树简介【详解】

为什么区间求和不用 for 循环累加?

直接遍历 arr[l]arr[r] 求和,时间复杂度是 O(n)。当有大量查询(比如 10⁵ 次)或数组频繁修改时,总耗时会爆炸。这时需要预处理结构——树状数组(Fenwick Tree)和线段树(Segment Tree)就是为此而生的两个主流选择。

std::vector 上手写树状数组:5 步实现单点更新 + 区间查询

树状数组代码短、常数小、易调试,适合只涉及「单点修改 + 区间求和」的场景。注意它下标必须从 1 开始,实际使用时常把原数组整体右移一位。

  • 定义 tree 数组大小为 n + 1n 是原数组长度)
  • lowbit(x) 写成 x & -x,别用循环或 pow
  • 更新操作:void add(int i, int delta),从 i 开始向上跳 i += lowbit(i)
  • 前缀和查询:int sum(int i),从 i 向下跳 i -= lowbit(i)
  • 区间 [l, r] 求和 = sum(r) - sum(l-1),注意 l 是原数组下标(已转为 1-indexed)
int n = arr.size(); vector tree(n + 1, 0); 

auto lowbit = [](int x) { return x & -x; };

auto add = [&](int i, int delta) { while (i <= n) { tree[i] += delta; i += lowbit(i); } };

auto sum = [&](int i) -> long long { long long res = 0; while (i > 0) { res += tree[i]; i -= lowbit(i); } return res; };

// 初始化:对每个位置 i(原下标 0 → 新下标 1) for (int i = 0; i < n; ++i) { add(i + 1, arr[i]); }

// 查询 [l, r](原下标,闭区间) long long range_sum = sum(r + 1) - sum(l);

线段树更适合什么场景?

当你不止要区间求和,还要支持「区间赋值」「区间加法」「最大值/最小值查询」「混合操作」时,线段树更灵活。但它代码长、内存开销大(约 4 * n)、容易写错边界。

立即学习C++免费学习笔记(深入)”;

  • 必须用数组模拟树:根节点下标为 1,左子节点是 2 * rt,右子是 2 * rt + 1
  • 懒标记(lazy)不是可选优化,而是支持区间更新的必要设计
  • 建树、查询、更新都要递归,终止条件是当前区间完全落在查询/更新范围内
  • 每次查询或更新后,记得在回溯时 push_up(合并子节点结果)
  • 如果只做单点更新,可以省略 lazy,但区间更新不加 lazy 会退化成 O(n)

树状数组 vs 线段树:几个关键差异点

别盲目套模板。选错结构会让调试时间翻倍。

  • 空间:树状数组仅需 O(n),线段树至少 O(4n),内存紧张时树状数组优势明显
  • 编码成本:树状数组核心逻辑 20 行内可写完;线段树带懒标记通常 > 80 行,且 push_downpush_up 顺序极易出错
  • 扩展性:树状数组难以支持区间最值、历史版本、二维扩展;线段树可改造成可持久化、动态开点、ZKW 形式等
  • 常数性能:同为区间求和,树状数组实测快 30%~50%,尤其在高频查询场景

真正容易被忽略的是:树状数组不能直接查区间最值,也不能高效支持「第 k 小」这类顺序统计问题——这时候哪怕多写 100 行,也得切到线段树或平衡树。

text=ZqhQzanResources