python利用前序和中序遍历构建二叉树的递归算法,并举例
时间: 2024-02-25 17:54:19 浏览: 78
根据先序与中序遍历结果建立二叉树
4星 · 用户满意度95%
好的,假设现在有一棵二叉树,它的前序遍历和中序遍历分别如下:
前序遍历:[3, 9, 20, 15, 7]
中序遍历:[9, 3, 15, 20, 7]
根据这两个遍历结果,我们可以使用递归算法构建出这棵二叉树,具体实现如下:
```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):
# 如果前序遍历或中序遍历为空,则返回 None
if not preorder or not inorder:
return None
# 根据前序遍历确定当前子树的根节点
root_val = preorder[0]
root = TreeNode(root_val)
# 根据中序遍历确定当前子树的左右子树
idx = inorder.index(root_val)
root.left = buildTree(preorder[1:idx+1], inorder[:idx])
root.right = buildTree(preorder[idx+1:], inorder[idx+1:])
return root
preorder = [3, 9, 20, 15, 7]
inorder = [9, 3, 15, 20, 7]
root = buildTree(preorder, inorder)
```
这样,我们就得到了一棵二叉树,它的结构如下:
```
3
/ \
9 20
/ \
15 7
```
阅读全文