树状数组单点修改需传增量delta而非目标值,通过i += i & -i向上更新父节点,如a[i]从5改为9应调用update(i, 4)。

单点修改怎么写 update 函数
树状数组的单点修改本质是「给位置 i 加上一个增量 delta,然后向上更新所有受影响的父节点」。关键不是重设值,而是传差值——比如要把 a[i] 从 5 改成 9,得调用 update(i, 4),而不是 update(i, 9)。
常见错误是把 update 当成“赋值”,结果整个前缀和崩掉。正确写法依赖 i += i & -i 循环跳转:
void update(int i, int delta) {
while (i
-
i从 1 开始计数(别用 0-indexed 数组直接套,容易越界) - 如果原数组是
vector<int> a</int>,修改第i个元素时,先算delta = new_val - a[i],再update(i, delta),最后更新a[i] = new_val - 注意
n是树状数组容量(通常等于原数组长度),不是tree.size()—— 如果tree开了n+1大小,i 才安全
区间查询为什么用两次 query 相减
树状数组原生只支持「前缀和查询」,即 query(r) 返回 [1..r] 的和。要查 [l, r],只能拆成 query(r) - query(l-1)。这个减法不是优化技巧,是定义使然——没有直接查区间的底层操作。
典型翻车点:
- 写成
query(r) - query(l):漏减左端点,结果少算a[l] - 对 0-indexed 原数组硬套公式:比如数组下标从 0 开始,但
query内部按 1-indexed 实现,此时查[l, r]应调用query(r+1) - query(l),不是直接塞l/r -
query边界越界:当l == 1,query(l-1)变成query(0),而标准实现里query(0)应返回 0(循环不进 while),否则要加特判
C++ 实现里 lowbit 怎么写才不出错
i & -i 是最常用也最稳妥的 lowbit 写法,它直接利用补码特性,比 1 更通用(后者要求 <code>i != 0,且非所有编译器都支持)。
组合式C++代码评审方案,融合静态分析、AI推理、多轮迭代评审和C++专项检查,适用于PR审查、增量代码审查、全项目评审和代码质量评分,触发词包括review cpp、cpp代码评审、C++review、代码审查。
必须确保 i 是正整数。如果传入 0,i & -i 得 0,while (i 会死循环。所以 <code>update 和 query 入口建议加断言或防护:
assert(i >= 1 && i
- 别用
unsigned int算-i:无符号数取负会回绕,i & -i失效 - 如果封装成类,把
tree定义为vector<long long></long>,避免多次修改后前缀和溢出(尤其题目没说数据范围时) -
query函数里循环是i > 0,不是i >= 1——因为i每次减去lowbit,最终会变成 0,i > 0才能自然退出
初始化时别直接 for 循环调 update
如果原数组是 a[1..n],想把初始值灌进树状数组,最慢但最安全的方式是 for (int i = 1; i 。时间复杂度 <code>O(n log n),对 n ≤ 1e6 可能卡常。
更优解是用「差分 + 树状数组」思想做 O(n) 初始化:
for (int i = 1; i
- 这个写法本质是模拟树状数组建树过程,跳过中间冗余更新
- 但前提是
tree初始全为 0,且你完全信任自己没写错边界(j 必须判断,否则越界) - 竞赛中若时间充裕,优先用多次
update——少一行代码少一个 bug;工程中若n极大且初始化频繁,再考虑线性建树
树状数组的陷阱不在原理多难,而在下标、差值、边界这三处细节反复咬人。哪怕抄对模板,只要改了索引方式或混用 0/1-indexed,就大概率在某个 l=1 或 i=n 的 case 上跪。
C++免费学习笔记(深入):立即使用
在学习笔记中,你将探索 C++ 的入门与实战技巧!










