python求二叉树中度为2的结点个数代码
时间: 2024-05-07 17:22:57 浏览: 111
下面是一个递归实现的示例代码:
```python
class TreeNode:
def __init__(self, val):
self.val = val
self.left = None
self.right = None
def count_nodes(root):
if not root:
return 0
count = 0
if root.left and root.right:
count = 1
count += count_nodes(root.left) + count_nodes(root.right)
return count
```
其中 `TreeNode` 是二叉树的节点类,`val` 是节点的值,`left` 和 `right` 分别是左子树和右子树。 `count_nodes` 函数递归地遍历二叉树,如果当前节点的左右子树均不为空,则该节点为度为2的节点,计数器加一,并继续递归遍历其左右子树。最后返回计数器的值即可。
使用示例:
```python
# 构建一棵二叉树
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)
root.right.right = TreeNode(7)
# 计算度为2的节点个数
count = count_nodes(root)
print(count) # 输出 3
```
在上面的示例中,输入的二叉树共有3个度为2的节点,即节点2、3和6。
阅读全文