要求实现一个函数,求二叉树中值为 key 的节点个数(输入-1表示指针为 NULL)下图所示二叉树中值为1的结点个数为3 个
时间: 2024-02-18 11:01:05 浏览: 24
以下是求二叉树中值为 key 的节点个数的 Python 代码实现:
```python
class TreeNode:
def __init__(self, val):
self.val = val
self.left = None
self.right = None
def count_node(root, key):
if not root:
return 0
count = 0
if root.val == key:
count = 1
count += count_node(root.left, key)
count += count_node(root.right, key)
return count
# 测试
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(1)
root.left.right = TreeNode(4)
root.right.left = TreeNode(1)
root.right.right = TreeNode(5)
print(count_node(root, 1)) # 输出 3
```
该函数的实现思路是递归地遍历二叉树,如果当前节点的值等于 key,则计数器 count 加1,然后递归遍历当前节点的左子树和右子树,将它们的计数器值累加到 count 中。最后返回 count 的值即可。