输入某二叉树的前序遍历和中序遍历的结果,请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
思路:
- 前序遍历 根节点-》左子树-》右子树
- 中序遍历 左子树-》根节点-》右子树
- 根据两个遍历结果中都不存在重复数字,那么前序遍历的第一个数值为起始点,找出该值在中序遍历数组中的位置,该位置左侧为左子树,右侧为右子树
递归函数:
- 终止条件:前序和中序数组为空
- 单次过程 : 根据前序数组确定根节点的值,根据根节点在中序数组的位置划分左右子树,然后递归构造左右子树
# Definition for a binary tree node.
# 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]) -> TreeNode:
if len(preorder) ==0 or len(inorder) == 0:
return None
root = TreeNode(preorder[0])
mid_index = inorder.index(preorder[0])
root.left = self.buildTree(preorder[1:mid_index+1],inorder[:mid_index])
root.right = self.buildTree(preorder[mid_index+1:],inorder[mid_index+1:])
return root