从先序与中序遍历序列构造二叉树
描述:
给定两个整数数组 preorder 和 inorder ,其中 preorder 是二叉树的先序遍历, inorder 是同一棵树的中序遍历,请构造二叉树并返回其根节点。
递归法
解题思路:
通过先序遍历我们可以找到root,因为列表中没有重复数字的原因,根据root我们可以根据中序找到root的索引位置,该索引位置左边就是左子树,右边就是右子树。而且根据中序中root的索引位置,可以在先序中确定先序里左子树索引位置为[1:1+idx),右子树索引位置[1+idx:-1]。根据位置,通过递归法直到遍历到结束。
ps:图片来源leetcode题解
代码实现:
class TreeNode:
def __init__(self,x):
self.val = x
self.left = None
self.right = None
class Solution:
def buildTree(self,preorder:List[int],inorder:List[int])->:
# 递归中止条件
if not preorder or not inorder:
return
# 根据先序列表得到根节点root
root = TreeNode(preorder[0])
# 根据中序列表得到根节点的索引位置
idx = inorder.index(preorder[0])
# 递归法找到根节点的左子树部分
root.left = self.buildTree(preorder[1:1+idx],inorder[0:idx])
# 递归法找到根节点的右子树部分
root.right = self.buildTree(preorder[1+idx:],inorder[idx+1:])
return root