递归统计节点数最简洁但易栈溢出,适用于树高可控场景;迭代法用栈模拟dfs,空间o(h)更稳定,推荐生产环境使用;morris遍历虽o(1)空间但修改原树且复杂,一般不推荐。

递归统计二叉树节点数:简洁但要注意栈溢出
直接用 root == nullptr 判空,返回 0;否则返回 1 + countNodes(root->left) + countNodes(root->right)。这是最自然的写法,逻辑清晰,代码量少。
但要注意:深度过大的树(比如退化成链表的单支树,高度 > 10⁵)会触发栈溢出。g++ 默认栈空间约 8MB,对应递归深度约 10⁴~10⁵ 级,实际取决于局部变量大小和编译器优化。
常见错误现象:Segmentation fault (core dumped) 或 stack overflow,尤其在 LeetCode 测试用例含极端偏斜树时。
- 使用场景:树高可控(如平衡树、教学示例)、调试阶段快速验证
- 性能影响:时间 O(n),空间 O(h),h 是树高
- 可加尾递归优化?C++ 编译器通常不优化这种二叉递归,
[[likely]]或inline无实质帮助
迭代统计节点数:用 stack 或 queue 都行,但 stack 更省空间
用 std::stack<treenode></treenode> 模拟递归过程,避免函数调用开销与栈限制。每次 pop 一个节点,计数加 1,再把非空子节点 push 进去。
相比 std::queue(BFS),std::stack(DFS 迭代)空间占用更小:最坏情况仅 O(h),而 BFS 迭代在满二叉树最后一层需 O(n/2) 空间。
示例关键片段:
组合式C++代码评审方案,融合静态分析、AI推理、多轮迭代评审和C++专项检查,适用于PR审查、增量代码审查、全项目评审和代码质量评分,触发词包括review cpp、cpp代码评审、C++review、代码审查。
int countNodes(TreeNode* root) {
if (!root) return 0;
std::stack<treenode> stk;
stk.push(root);
int cnt = 0;
while (!stk.empty()) {
TreeNode* node = stk.top(); stk.pop();
cnt++;
if (node->right) stk.push(node->right);
if (node->left) stk.push(node->left);
}
return cnt;
}</treenode>
- 使用场景:生产环境、树结构不可控、需稳定内存占用
- 注意点:子节点入栈顺序影响遍历方向,但不影响总数,这里先右后左等价于递归的“根-左-右”顺序
- 性能影响:时间仍是 O(n),空间 O(h),但 h 是运行时栈深,不是系统调用栈,更可控
性能对比实测的关键变量:树形态比语言本身影响更大
别只看“递归 vs 迭代”标签。真正拉大差距的是树的形状——在完全平衡树上,两者时间差不到 10%;但在单支链表状树上,递归可能直接崩溃,而迭代稳稳跑完。
实测建议用 std::chrono::high_resolution_clock 计时,但必须关闭编译器优化(-O0)才能看出差异;开 -O2 后,递归版可能被部分内联,反而略快(因无容器分配/销毁开销)。
- 容易踩的坑:用
std::vector当栈(push_back/pop_back)比std::stack稍慢但更灵活;误用std::deque做 BFS 且未预留空间,导致多次 realloc - 参数差异:迭代版可轻松改造成带 early-return 的变体(如计数超阈值就停),递归版难做这类中断
- 兼容性:C++11 起所有标准容器都可用,无需额外依赖
要不要用 Morris 遍历?理论 O(1) 空间但实际不推荐
Morris 遍历确实能把空间压到 O(1),靠临时修改树的 right 指针建线索,遍历完再还原。但问题在于:TreeNode 通常来自外部(如 LeetCode 输入),修改它可能违反 const 正确性或引发多线程问题;且代码复杂度陡增,debug 成本高。
除非你明确受限于嵌入式设备的几 KB 内存,否则为省那点栈空间不值得。现代服务器或本地开发机,O(h) 迭代栈空间(通常几百字节)远不如一次 new 分配的开销敏感。
- 真实瓶颈往往不在遍历本身,而在节点构造、内存分配器争用或缓存不友好访问模式
- 如果树是静态的、只读的,且已知最大高度,甚至可以预分配固定大小数组模拟栈,比
std::stack更快
实际项目里,优先选迭代版 —— 它不炫技,不出错,也足够快。递归版留着写原型或讲算法原理用。Morris 就当了解即可,真遇到 O(1) 空间需求,先确认是不是真的不能用栈,而不是一上来就啃硬骨头。
C++免费学习笔记(深入):立即使用
在学习笔记中,你将探索 C++ 的入门与实战技巧!










