C++怎么手写树状数组(Fenwick Tree)

风明大大_7720

风明大大_7720

2026-11-03

263人浏览

原创

不能直接用 std::vector 初始化树状数组,因为需手动实现lowbit运算、1-based索引、update/query逻辑;必须用x & -x计算lowbit,初始化需逐调update,tree[0]不用,大小设为n+1。

c++怎么手写树状数组(fenwick tree)

为什么不能直接用 std::vector 初始化就完事?

手写树状数组不是套个容器就行——核心在于 lowbit 运算和下标偏移处理。C++ 标准库没提供现成的 Fenwick Tree,你得自己管好:数组从 1 开始索引、更新时向上跳、查询时向下拆解。常见错误是把下标当 0 起始来写 update 或 query,结果越界或漏加。

推荐用 std::vector 存值(防 int 溢出),大小设为 n + 1,tree[0] 永远不用。

  • lowbit(x) 必须写成 x & -x,别用循环或 __builtin_ctz(后者依赖平台且不通用)
  • 初始化时逐个调用 update(i, val),别直接赋值到 tree[i] —— 否则父子关系没建起来
  • 如果原数组叫 a,建树代码就是:
    for (int i = 1; i 

update 和 query 的边界怎么写才不崩?

两个函数都只操作合法下标:1 到 n。写错 while 条件会导致死循环或跳过根节点。关键点是「更新往上走,查询往下拆」。

  • update(int i, int delta):从 i 开始,每次加 lowbit(i),直到超过 n
  • query(int i):从 i 开始,每次减 lowbit(i),直到变成 0(注意不是 1)
  • 区间查询 [l, r] 写成 query(r) - query(l - 1),这里 l - 1 可能为 0,但 query(0) 返回 0 是安全的(因为 while 不进)

示例片段:

void update(int i, int delta) {<br>  while (i     tree[i] += delta;<br>    i += i & -i;<br>  }<br>}<br><br>long long query(int i) {<br>  long long s = 0;<br>  while (i > 0) {<br>    s += tree[i];<br>    i -= i & -i;<br>  }<br>  return s;<br>}

单点改 + 区间查 vs 区间改 + 单点查,要不要换结构?

纯手写树状数组默认支持「单点更新 + 前缀和查询」,也就是单点改、区间查。想支持「区间加 + 单点查」,得用差分 + 树状数组,即维护差分数组 d,然后 update(l, v), update(r+1, -v);单点查就是 query(i)。

C++14
C++14

C++14 对 C++11 的修正与增强版本,适合旧系统维护和较老工具链兼容。

下载

但如果要「区间加 + 区间和」,单树状数组搞不定,必须上两棵树(常见套路:一棵维护 d[i],一棵维护 i * d[i]),或者直接换线段树。别硬撑——Fenwick 的优势是短小快,不是功能全。

  • 确认题型再动手:90% 的 OJ 树状数组题都是单点改 + 区间查
  • 差分版本的 update 逻辑不变,只是调用方式变:add(l, v); add(r+1, -v);
  • 别在同一个 tree 上混用两种模式,容易符号搞反、下标越界

模板里要不要加 static 或封装成类?

竞赛场景下直接写全局函数+全局 vector 最快;但多人协作或多次复用时,封装成类更稳。重点不是语法糖,而是避免状态残留——比如没清空 tree 就重用,或者 n 变了但 tree.size() 没重置。

  • 类封装时,构造函数必须接受 n 并 resize:tree.resize(n + 1)
  • 不要把 lowbit 写成成员函数,它只是个 inline 工具,定义成 constexpr auto lowbit = [](int x) { return x & -x; }; 更干净
  • 如果用 C++17,可加 [[nodiscard]] 在 query 上,防止忘了接返回值

最简可用类骨架:

struct Fenw {<br>  int n;<br>  std::vector<long long> tree;<br>  Fenw(int sz) : n(sz), tree(sz + 1) {}<br>  void update(int i, long long delta) { /* 同上 */ }<br>  long long query(int i) { /* 同上 */ }<br>};</long>

真正容易被忽略的是:每次 new 一个 Fenw 实例,都要确保 n 和实际数据范围一致;否则 update 里的 i 判断会提前截断。

C++免费学习笔记(深入):立即使用
在学习笔记中,你将探索 C++ 的入门与实战技巧!

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
c++和c语言的区别有哪些
c++和c语言的区别有哪些

c++和c语言的区别:1、面向对象编程(OOP)支持不同;2、新增特性不同;3、标准库不同;4、编译方式不同;5、命名空间不同等等。想了解更多c++的相关内容,可以阅读本专题下面的文章。

2024.03.14

2288

9

c++和python学习顺序推荐
c++和python学习顺序推荐

一般建议先学习C++,再学习Python,因为这样可以逐步从较为底层的编程语言向更高级的语言过渡。想了解更多python的相关内容,可以阅读本专题下面的文章。

2024.03.14

1039

6

python和c++学习性价比分析
python和c++学习性价比分析

Python易于学习,广泛应用于Web开发、数据科学和人工智能等领域,但性能较低。C语言性能高,适用于对性能要求较高的场景,如游戏开发和系统编程,但学习曲线陡峭,错误处理复杂。想了解更多python的相关内容,可以阅读本专题下面的文章。

2024.03.14

447

5

c语言和c++一样吗
c语言和c++一样吗

c语言和c++是两种不同的编程语言,虽然有相似之处,但存在显著差异。c语言专注于过程式编程和系统级开发,以简洁、高效著称。c++作为c语言的超集,引入了面向对象编程,增强了代码组织和管理能力,但学习曲线也更陡峭。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.03.14

347

5

c语言和c++先学哪个好
c语言和c++先学哪个好

初学者选择学习c语言还是c++语言,需要根据个人学习目标、背景以及编程兴趣和预期应用方向来决定。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.03.14

426

5

c语言和c++的区别和联系
c语言和c++的区别和联系

c语言和c++是计算机科学领域应用广泛的编程语言。虽然它们有着相似的基础,但它们在语言类型、语法功能和内存管理方面存在着显著差异。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.03.14

600

5

c++软件中文更改教程
c++软件中文更改教程

对于 ide,可通过打开设置,找到语言设置,选择中文,并保存更改。对于非 ide 应用程序,可查找设置或选项,选择语言设置,更改为中文,并保存更改。想了解更多c++的相关内容,可以阅读本专题下面的文章。

2024.03.21

1429

9

python和java和c++学习性价比分析
python和java和c++学习性价比分析

Python以其易学性、丰富的库和活跃的社区而著称,适合数据科学、人工智能和Web开发。Java以其跨平台性、企业级应用开发和Android应用开发而闻名。C++以其底层控制能力、高效性能和游戏开发而著称。选择哪种语言取决于个人兴趣、职业方向和特定需求。想了解更多python和java和c++的相关内容,可以阅读本专题下面的文章。

2024.03.22

1237

7

c++和c语言学习顺序推荐
c++和c语言学习顺序推荐

对于初学者,建议先学习C语言,掌握编程基础后再转入C++,便于理解面向对象编程概念。有编程经验者可直接学习C++,快速接触高级编程技术。想了解更多c++和c语言的相关内容,可以阅读本专题下面的文章。

2024.03.25

1325

9

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
C++ Core Guidelines
C++ Core Guidelines

共0课时 | 0人学习

C++ Reference
C++ Reference

共0课时 | 0人学习

C++ 官方标准说明
C++ 官方标准说明

共0课时 | 0人学习