递归法最简洁但易栈溢出,适用于树高可控场景;迭代栈法需判空防崩溃;Morris遍历O(1)空间但会临时改树结构,适合超大深度且允许修改的场景。

递归统计二叉树节点数:简洁但要注意栈溢出
直接用 root == nullptr 判空,返回 0;否则返回 1 + countNodes(root->left) + countNodes(root->right)。这是最自然的写法,逻辑清晰,代码三行搞定。
但要注意:深度过大的树(比如退化成链表的单支树,高度 > 10⁵)会导致栈溢出。g++ 默认栈空间通常只有 8MB,递归调用深度超过几万就容易崩。
常见错误现象:Segmentation fault (core dumped) 或程序直接退出,没报错信息——十有八九是爆栈了,不是空指针解引用。
使用场景:开发调试、树高可控(如 AVL/红黑树实测场景)、LeetCode 中数据规模明确 ≤ 10³ 的题目。
立即学习“C++免费学习笔记(深入)”;
迭代法用栈模拟:避免爆栈,但要小心重复入栈
手动维护一个 std::stack<treenode></treenode>,初始 push root,每次 pop 一个节点,计数器加 1,再把它的非空左右子节点 push 进去。
关键点在于:必须检查 node != nullptr 再 push,否则栈里混入空指针,后续 pop 后解引用 node->left 就会崩溃。
容易踩的坑:
- 忘记判空就 push
root,当输入为nullptr时,第一次 pop 得到空指针,接着访问->left→segmentation fault - 先 push 左右再计数,导致空节点也被计 —— 实际上空指针根本不该进栈
- 用
queue做 BFS 也能统计,但空间占用和栈版无本质区别,只是遍历顺序不同,不影响总数
迭代法用 Morris 遍历:O(1) 空间但会临时改树结构
Morris 遍历能在 O(n) 时间、O(1) 额外空间下遍历二叉树,自然也能用来计数。它通过临时复用叶子节点的空右指针,构建线索回溯路径。
优点是内存零开销,适合嵌入式或超大深度树(比如日志系统中深度达百万的解析树);缺点是会短暂修改原树指针(虽然后续会恢复),若树被多线程访问或有 const 语义要求,就不能用。
参数差异明显:countNodesMorris(TreeNode* root) 内部没有递归调用,也没有 stack/queue,只靠几个指针变量(cur, pre)和 while 循环推进。
性能影响:比普通迭代略慢(多了指针修复操作),但空间从 O(h) 降到 O(1),h 是树高。当 h ≈ n 时,优势极大。
什么时候该选哪种?看运行环境和树形态
本地小数据快速验证,用递归最省事;服务器长期运行且树深度不可控,优先迭代栈版;资源极度受限又允许改树,才考虑 Morris。
别忽略一个细节:C++ 中 TreeNode 结构体本身是否含智能指针(如 std::unique_ptr<treenode></treenode>)会影响迭代时的取址方式——用 get() 拿裸指针,别直接 push 智能指针对象,否则栈里存的是副本,析构后指针失效。


















