讲师中心 微信公众号
AI工具推荐 视频效率加速

C++如何实现二叉树的节点总数计算

轻涛小哥_5783

轻涛小哥_5783

发布时间:2026-06-18 17:51:42

|

553人浏览过

|

来源于php中文网

原创

递归计算二叉树节点总数的核心是“当前节点+左子树节点数+右子树节点数”,需先判空再累加1,时间复杂度O(n),空间复杂度O(h);迭代法用栈模拟,避免栈溢出;完全二叉树可优化至O(log²n)。

c++如何实现二叉树的节点总数计算

递归计算节点总数最直接

二叉树节点总数本质上是“当前节点 + 左子树节点数 + 右子树节点数”,递归天然匹配这个结构。只要树不为空,就累加 1 再递归左右子树;空节点返回 0。

常见错误是忘记判空导致 nullptr 解引用崩溃,或漏掉当前节点的 +1:

int countNodes(TreeNode* root) {
    if (!root) return 0;              // 必须先判空
    return 1 + countNodes(root->left) + countNodes(root->right);
}
  • TreeNode* 是典型节点指针类型,实际使用时需确认你的节点结构名(如 Node* 或 BSTNode*)
  • 时间复杂度 O(n),空间复杂度 O(h)(h 为树高,最坏退化成链表时为 O(n))
  • 不要写成 return countNodes(root->left) + countNodes(root->right) —— 这会漏掉根节点

迭代方式用栈模拟递归

当树很深、担心栈溢出时,改用显式栈避免系统调用栈压栈。核心逻辑不变:每弹出一个节点,计数器加 1,再把它的非空子节点压入栈。

容易踩的坑是子节点入栈顺序不重要,但必须检查是否为 nullptr,否则会往栈里塞无效指针:

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

C++ Code Review Master
C++ Code Review Master

组合式C++代码评审方案,融合静态分析、AI推理、多轮迭代评审和C++专项检查,适用于PR审查、增量代码审查、全项目评审和代码质量评分,触发词包括review cpp、cpp代码评审、C++review、代码审查。

下载
int countNodes(TreeNode* root) {
    if (!root) return 0;
    stack<TreeNode*> s;
    s.push(root);
    int count = 0;
    while (!s.empty()) {
        TreeNode* node = s.top(); s.pop();
        count++;  // 当前节点计入
        if (node->left) s.push(node->left);
        if (node->right) s.push(node->right);
    }
    return count;
}
  • 用 stack 而不是 queue——这里不需要层序遍历,DFS/BFS 都能数清节点,选栈更贴近递归语义
  • 如果用 queue,只需把 stack 换成 queue,top() 改成 front(),pop() 前加 pop()
  • 迭代写法空间复杂度仍是 O(h)(栈中最多存一条路径上的节点),不是 O(1)

完全二叉树可优化到 O(log²n)

如果题目明确说“完全二叉树”(complete binary tree),就不能套用通用方法——利用其结构规律:除最后一层外全满,且最后一层靠左。此时可通过比较左右子树高度,决定哪边是满二叉树,跳过部分遍历。

关键点在于:满二叉树节点数 = 2^h - 1,而完全二叉树的左右子树至多差一层:

int countNodes(TreeNode* root) {
    if (!root) return 0;
    TreeNode* left = root, *right = root;
    int heightLeft = 0, heightRight = 0;
    while (left) { left = left->left; heightLeft++; }
    while (right) { right = right->right; heightRight++; }
    if (heightLeft == heightRight)
        return (1 << heightLeft) - 1; // 满二叉树:2^h - 1
    return 1 + countNodes(root->left) + countNodes(root->right);
}
  • 必须严格满足完全二叉树定义才适用,普通二叉树用这个反而更慢
  • 1 等价于 <code>pow(2, heightLeft),但位运算更快且避免浮点误差
  • 每次递归只走一边,总时间复杂度 O(log²n),因为每层最多做一次 O(log n) 的高度探测

带计数成员变量的树类设计

如果频繁查询节点总数(比如在插入/删除后实时维护),硬算每次都是 O(n),不如在树结构里缓存总数。这时需保证所有修改操作同步更新计数器。

典型错误是只在插入时加 1,却忘了删除时减 1,或子树旋转、合并等操作后未重算:

struct BST {
    TreeNode* root;
    int size; // 缓存节点总数
    BST() : root(nullptr), size(0) {}
    void insert(int val) {
        root = insertHelper(root, val);
        size++; // 插入成功才加
    }
    void erase(int val) {
        if (find(val)) {
            root = eraseHelper(root, val);
            size--; // 删除成功才减
        }
    }
};
  • size 是易错点:必须与所有变更操作严格耦合,否则数值迅速失真
  • 适用于读多写少场景;若写操作极频繁,维护开销可能抵消查询收益
  • 序列化/反序列化时别忘了保存和恢复 size 字段,否则重建后计数归零

实际用哪个方法,取决于树的形态、调用频率和是否允许修改结构。通用场景优先写递归,深度隐患明显时换迭代,确定是完全二叉树再上高度探测,高频查询且可控写入才考虑缓存 size。

热门AI工具

更多
超级简历WonderCV

一款AI办公效率工具,主要用于免费求职简历模版下载制作,应届生职场人必备简历制作神器,适合需要提升相关任务效率的用户。

UP简历
UP简历 Hot

一款AI办公效率工具,主要用于基于AI技术的免费在线简历制作工具,适合需要提升相关任务效率的用户。

AionClaw
AionClaw Hot

AionClaw是一款面向办公、创作和编程任务的AI桌面智能体。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

VibeKnow
VibeKnow Hot

一款AI视频创作工具,主要用于全球首个AI知识视频创作平台,文档、文章、网页,一键生成视频,适合需要提升相关任务效率的用户。

PixPix
PixPix Hot

PixPix是一款面向电商视觉生产的AI商品图生成工具。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

Atoms
Atoms Hot

Atoms是一款AI智能体工具,第一支自动构建真实业务的 AI 团队。

相关专题

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

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

2248

2024.03.14

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

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

1019

2024.03.14

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

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

427

2024.03.14

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

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

327

2024.03.14

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

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

406

2024.03.14

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

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

580

2024.03.14

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

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

1409

2024.03.21

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

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

1217

2024.03.22

VibeKnow知识视频制作与导出详解
VibeKnow知识视频制作与导出详解

本专题整理VibeKnow从注册、素材上传、AI生成到预览导出的完整操作,帮助新手快速制作第一条知识讲解视频。

0

2026.10.14

热门下载

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

精品课程

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

共0课时 | 0人学习

C++ Reference
C++ Reference

共0课时 | 0人学习

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

共0课时 | 0人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn