答案是使用递归或迭代方法计算二叉树节点数:递归通过左右子树相加加1实现,迭代用栈模拟遍历计数。

在C++中计算二叉树的节点数量,通常使用递归方法。核心思想是:当前节点的数量等于左子树节点数 + 右子树节点数 + 1(当前节点本身)。
定义二叉树节点结构
首先需要定义二叉树的节点结构,一般如下:
struct TreeNode {
int val;
TreeNode* left;
TreeNode* right;
TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
};
递归计算节点数量
通过递归遍历整棵树,统计所有节点。如果节点为空,返回0;否则返回左右子树节点总数加1。
算家云
37
高效、便捷的人工智能算力服务平台
37
查看详情
int countNodes(TreeNode* root) {
if (root == nullptr) {
return 0;
}
return countNodes(root->left) + countNodes(root->right) + 1;
}
非递归方法(使用栈模拟遍历)
也可以用迭代方式避免递归带来的栈溢出风险,使用栈或队列进行层序或前序遍历,每访问一个节点计数加1。
立即学习“C++免费学习笔记(深入)”;
#include <stack>
int countNodesIterative(TreeNode* root) {
if (root == nullptr) return 0;
std::stack<TreeNode*> stk;
stk.push(root);
int count = 0;
while (!stk.empty()) {
TreeNode* node = stk.top();
stk.pop();
if (node) {
count++;
stk.push(node->right);
stk.push(node->left);
}
}
return count;
}
基本上就这些。递归写法简洁易懂,适合大多数场景;迭代方法更安全,适合深度较大的树。根据实际需求选择即可。
相关标签:
node 栈 c++ 递归 栈
大家都在看:
C++堆和栈内存分配区别
C++异常处理与堆栈展开机制解析
C++内存管理基础中堆内存和栈内存的区别
C++STL栈stack操作与应用实例
C++栈内存与堆内存区别与应用
C++异常处理与堆栈展开机制解析
C++内存管理基础中堆内存和栈内存的区别
C++STL栈stack操作与应用实例
C++栈内存与堆内存区别与应用
本站资料仅供学习交流使用请勿商业运营,严禁从事违法,侵权等任何非法活动,否则后果自负!
THE END
































暂无评论内容