商城首页欢迎来到中国正版软件门户

您的位置:首页 >C++统计二叉树叶子节点方法

C++统计二叉树叶子节点方法

  发布于2025-12-14 阅读(0)

扫一扫,手机访问

统计二叉树叶子节点需遍历树并判断节点左右子树为空;2. 可用递归或非递归方法实现;3. 递归法代码简洁,迭代法用栈避免深度过大导致的栈溢出。

c++中如何统计二叉树叶子节点_c++二叉树叶子节点统计方法

在C++中统计二叉树的叶子节点,核心思路是遍历整棵树,对每个节点判断是否为叶子节点。叶子节点的定义是:左右子节点都为空的节点。

定义二叉树节点结构

通常使用结构体来表示二叉树的节点:

struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
};

递归方法统计叶子节点

递归是最直观的方式。从根节点开始,如果当前节点是叶子节点,返回1;否则递归计算左子树和右子树的叶子节点数并相加。

int countLeaves(TreeNode* root) {
    if (root == nullptr) {
        return 0;
    }
    if (root->left == nullptr && root->right == nullptr) {
        return 1; // 当前节点是叶子
    }
    return countLeaves(root->left) + countLeaves(root->right);
}

非递归方法(使用栈)

也可以用栈模拟深度优先遍历,避免递归带来的函数调用开销。

#include <stack>
int countLeavesIterative(TreeNode* root) {
    if (!root) return 0;
    std::stack<TreeNode*> stk;
    stk.push(root);
    int count = 0;

    while (!stk.empty()) {
        TreeNode* node = stk.top();
        stk.pop();

        if (node->left == nullptr && node->right == nullptr) {
            count++;
        }

        if (node->left) stk.push(node->left);
        if (node->right) stk.push(node->right);
    }
    return count;
}

基本上就这些。递归写法简洁易懂,适合大多数场景;迭代方式更可控,适合树很深时防止栈溢出。根据实际需求选择即可。

本文转载于:互联网 如有侵犯,请联系zhengruancom@outlook.com删除。
免责声明:正软商城发布此文仅为传递信息,不代表正软商城认同其观点或证实其描述。

热门关注