JZ7 重建二叉树
描述:
给定节点数为 n 二叉树的前序遍历和中序遍历结果,请重建出该二叉树并返回它的头结点。
例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建出如下图所示。
提示:
-
vin.length == pre.length
-
pre 和 vin 均无重复元素
-
vin出现的元素均出现在 pre里
-
只需要返回根结点,系统会自动输出整颗树做答案对比
数据范围:n \le 2000n≤2000,节点的值 -10000 \le val \le 10000−10000≤val≤10000
要求:空间复杂度 O(n)O(n),时间复杂度 O(n)O(n)
思路: pre找根节点,将vin分成左子树和右子树,进行递归遍历
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def reConstructBinaryTree(self , pre: List[int], vin: List[int]) -> TreeNode:
if not pre : return None
root = TreeNode(pre[0])
index = vin.index(pre[0])
root.left = self.reConstructBinaryTree(pre[1:index+1], vin[:index])
root.right = self.reConstructBinaryTree(pre[index+1:], vin[index+1:])
return root