计算二叉树的高度、总节点个数和叶子节点个数代码
时间: 2024-01-26 10:05:10 浏览: 32
好的,下面是计算二叉树高度、总节点个数和叶子节点个数的 Python 代码:
```
class TreeNode:
def __init__(self, val):
self.val = val
self.left = None
self.right = None
def get_height(root):
if not root:
return 0
return max(get_height(root.left), get_height(root.right)) + 1
def get_node_count(root):
if not root:
return 0
return 1 + get_node_count(root.left) + get_node_count(root.right)
def get_leaf_count(root):
if not root:
return 0
if not root.left and not root.right:
return 1
return get_leaf_count(root.left) + get_leaf_count(root.right)
# 测试代码
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)
print(get_height(root)) # 输出 3
print(get_node_count(root)) # 输出 7
print(get_leaf_count(root)) # 输出 4
```
这里我们定义了一个 `TreeNode` 类来表示二叉树的节点,然后分别实现了 `get_height`、`get_node_count` 和 `get_leaf_count` 函数来计算二叉树的高度、总节点个数和叶子节点个数。测试代码中创建了一个简单的二叉树,并输出了计算结果。
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)