编写一个程序,实现二叉树高度的求取。利用二叉树的性质及二叉树前序遍历的思想,计算二叉树的高度。
时间: 2023-11-14 10:22:55 浏览: 79
以下是 Python 代码实现:
```python
class TreeNode:
def __init__(self, val):
self.val = val
self.left = None
self.right = None
def get_tree_height(root):
if not root:
return 0
left_height = get_tree_height(root.left)
right_height = get_tree_height(root.right)
return max(left_height, right_height) + 1
# 示例
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)
root.right.right = TreeNode(6)
print(get_tree_height(root)) # 输出 3
```
以上代码实现了一个二叉树节点类 `TreeNode` 和一个函数 `get_tree_height`,用于计算二叉树的高度。函数 `get_tree_height` 采用递归的方式,分别计算左子树和右子树的高度,并取两者之中的较大值,加上当前节点的高度 1 即可得到整个二叉树的高度。在示例中,二叉树的高度为 3。
阅读全文