二叉树高度python
时间: 2024-08-06 10:01:36 浏览: 56
二叉树的高度是指从根节点到最远叶子节点的最长路径上的边数。在Python中,计算二叉树的高度通常涉及到递归的方法。以下是简单的步骤:
1. 如果二叉树为空,则其高度为0;
2. 否则,递归地计算左子树和右子树的高度。取两者的最大值加1作为当前节点的高度。
下面是一个基本的Python函数实现:
```python
class TreeNode:
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
def height(root):
if root is None:
return 0
else:
# 计算左子树和右子树的高度
left_height = height(root.left)
right_height = height(root.right)
# 返回两者中的较大者加1
return max(left_height, right_height) + 1
```
你可以通过创建一个`TreeNode`实例并调用`height`函数来获取其高度。例如:
```python
# 创建一个简单的二叉树结构
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
# 计算高度
tree_height = height(root)
print("Binary tree height:", tree_height)
阅读全文