通过二叉树先序和中序遍历构建二叉树
时间: 2023-11-14 09:12:42 浏览: 93
二叉树的构建可以通过递归实现。具体步骤如下:
1. 从先序遍历中选择第一个节点作为根节点。
2. 在中序遍历中找到根节点的位置,将中序遍历分为左子树和右子树。
3. 根据左子树的长度,可以在先序遍历中找到左子树的先序遍历,右子树同理。
4. 递归地构建左子树和右子树。
代码实现如下:
```python
class TreeNode:
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
def buildTree(preorder, inorder):
if not preorder or not inorder:
return None
root_val = preorder[0]
root = TreeNode(root_val)
root_index = inorder.index(root_val)
left_inorder = inorder[:root_index]
right_inorder = inorder[root_index+1:]
left_preorder = preorder[1:1+len(left_inorder)]
right_preorder = preorder[1+len(left_inorder):]
root.left = buildTree(left_preorder, left_inorder)
root.right = buildTree(right_preorder, right_inorder)
return root
```
其中,preorder 和 inorder 分别为先序遍历和中序遍历的数组。在每一次递归中,我们取出先序遍历的第一个节点作为根节点,然后在中序遍历中找到该节点的位置,将中序遍历分为左子树和右子树。根据左子树的长度,我们可以在先序遍历中找到左子树的先序遍历,右子树同理。然后递归地构建左子树和右子树,并将它们连接到根节点上。最后返回根节点即可。
阅读全文