leetcode_105_从前序与中序遍历序列构造二叉树

从前序与中序遍历序列构造二叉树

描述

中等

根据一棵树的前序遍历与中序遍历构造二叉树。

注意:
你可以假设树中没有重复的元素。

示例

前序遍历 preorder = [3,9,20,15,7]

中序遍历 inorder = [9,3,15,20,7]

返回如下的二叉树：

   3
  / \
  9  20
    /  \
   15   7

解题

前序遍历第一个点肯定是根节点，然后在中序遍历中找到根节点，确认根节点左子树和右子树的节点

节点分布

• 前序遍历 [根节点 左子树节点 右子树节点]
• 中序遍历 [左子树节点 根节点 右子树节点]

按照前序和中序的节点分布，根节点的左子树部分和右子树部分已经得到了

左/右子树部分的前序和中序仍然符合上述的节点分布

递归建树

# 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 not preorder:
            return None
        # 前序遍历第一个节点为根节点
        x = preorder[0]
        root = TreeNode(x)
        # 找到中序遍历中该节点的位置
        index = inorder.index(x)
		# 递归建树
        root.left = self.buildTree(preorder[1:index+1], inorder[:index])
        root.right = self.buildTree(preorder[index+1:], inorder[index+1:])

        return root