求一个二叉树的叶子结点个数的函数
时间: 2024-05-15 14:16:01 浏览: 102
以下是 Python 代码实现:
```
class TreeNode:
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
def count_leaves(root: TreeNode) -> int:
if not root:
return 0
if not root.left and not root.right:
return 1
return count_leaves(root.left) + count_leaves(root.right)
```
其中,`TreeNode` 是二叉树节点的定义,`count_leaves` 函数接受一个根节点作为参数,返回二叉树中叶子节点的个数。如果根节点为空,则返回 0;如果根节点没有左右子节点,说明它是叶子节点,返回 1;否则递归计算左右子树的叶子节点个数之和。
相关问题
求一个二叉树的叶子结点个数的C加加函数
int countLeaf(TreeNode* root) {
if(root == NULL) {
return 0;
}
if(root->left == NULL && root->right == NULL) {
return 1;
}
return countLeaf(root->left) + countLeaf(root->right);
}
求二叉树叶子结点个数
求二叉树叶子节点的个数,可以通过递归遍历二叉树,然后统计叶子节点的个数来实现。具体实现可以参考以下代码:
``` python
# 定义二叉树结构体
class TreeNode:
def __init__(self, val):
self.val = val
self.left = None
self.right = None
# 定义统计叶子节点个数的函数
def count_leaf_node(root):
if not root:
return 0
if not root.left and not root.right:
# 如果当前节点没有左右子节点,说明当前节点是叶子节点
return 1
# 递归遍历左右子树,统计叶子节点个数
return count_leaf_node(root.left) + count_leaf_node(root.right)
# 测试样例
if __name__ == '__main__':
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)
root.right.left = TreeNode(6)
print(count_leaf_node(root)) # 输出:3
```
在上述代码中,我们定义了一个二叉树结构体 `TreeNode`,其中包含节点的值 `val`,以及左右子节点 `left` 和 `right`。接着,我们定义了一个函数 `count_leaf_node`,这个函数用来统计二叉树的叶子节点个数。在函数中,我们首先判断当前节点是否为空,如果为空,则返回 0。然后,我们判断当前节点是否为叶子节点,如果是,则返回 1。最后,我们递归遍历当前节点的左右子树,并累加左右子树的叶子节点个数,最终得到二叉树的叶子节点个数。在最后的测试样例中,我们构建了一个二叉树,并统计了它的叶子节点个数,输出结果为 3。
阅读全文