python利用前序和中序遍历构建二叉树的递归算法
时间: 2023-11-22 20:04:57 浏览: 89
构建二叉树的递归算法可以分为两个步骤:
1. 根据前序遍历确定当前子树的根节点。
2. 根据中序遍历确定当前子树的左右子树。
具体实现如下:
```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` 和 `inorder` 分别表示前序遍历和中序遍历的结果,返回构建好的二叉树的根节点。
阅读全文