给定一棵二叉树的先序遍历序列和中序遍历序列python代码实现
时间: 2023-12-27 11:04:41 浏览: 156
以下是使用Python实现给定一棵二叉树的先序遍历序列和中序遍历序列的代码:
```python
class TreeNode:
def __init__(self, val):
self.val = val
self.left = None
self.right = None
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)
root.left = buildTree(preorder[1: 1+root_index], inorder[:root_index])
root.right = buildTree(preorder[1+root_index:], inorder[root_index+1:])
return root
```
这个函数的输入参数包括先序遍历序列和中序遍历序列,返回值是构建出的二叉树的根节点。在函数内部,首先检查先序遍历序列和中序遍历序列是否为空。如果有任何一个为空,则返回None。否则,我们可以确定根节点的值为先序遍历序列的第一个元素root_val。接下来,我们在中序遍历序列中找到root_val的索引位置root_index,为了构建二叉树,我们将中序遍历序列分成两半:inorder[:root_index]是左子树的中序遍历序列,inorder[root_index+1:]是右子树的中序遍历序列。对应地,通过切片操作,我们将先序遍历序列分为两个部分:preorder[1:1+root_index]是左子树的先序遍历序列,preorder[1+root_index:]是右子树的先序遍历序列。最后,我们递归地构建左子树和右子树,并将其分别赋值给根节点的left和right属性。整个过程结束后,我们返回根节点root。
阅读全文