定义二叉树的数据结构,创建一颗二叉树,先序遍历该二叉树,计算二叉树的高度
时间: 2023-11-14 20:12:56 浏览: 119
数据结构二叉树的创建与遍历
好的,我来回答您的问题。
定义二叉树的数据结构,我们可以使用如下的代码实现:
```python
class TreeNode:
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
```
其中,每个节点包含三个属性:val表示该节点的值,left表示该节点的左子树,right表示该节点的右子树。
接下来我们可以使用如下的方式创建一颗二叉树:
```python
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)
```
这颗二叉树的结构如下图所示:
```
1
/ \
2 3
/ \
4 5
```
接下来我们可以使用先序遍历的方式遍历该二叉树,并输出每个节点的值:
```python
def preorder_traversal(root):
if root is None:
return
print(root.val)
preorder_traversal(root.left)
preorder_traversal(root.right)
preorder_traversal(root)
```
输出结果为:1 2 4 5 3。
最后,我们可以计算该二叉树的高度:
```python
def height(root):
if root is None:
return 0
return max(height(root.left), height(root.right)) + 1
print(height(root))
```
输出结果为:3。
阅读全文