【剑指offer】007. 重建二叉树

题目

【剑指offer】007. 重建二叉树
输入某二叉树的前序遍历和中序遍历的结果，请构建该二叉树并返回其根节点。
假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
示例 1:

Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
Output: [3,9,20,null,null,15,7]
示例 2:

Input: preorder = [-1], inorder = [-1]
Output: [-1]

限制：
0 <= 节点个数 <= 5000

解法

解法1

递归

class Solution:
    def buildTree(self, preorder: List[int], inorder: List[int]) -> TreeNode:
      # 每次取前序遍历的第一个数字，作为当前的根节点，用这个数字去中序遍历的结果里面找，记录在中序中的位置是idx,左边是左子树的集合
      if len(preorder)==0:
        return None
      root = TreeNode(preorder[0])
      if len(preorder)==1:
        return root

      idx = inorder.index(preorder[0]) # 左子树节点数量
      root.left = self.buildTree(preorder[1:idx+1], inorder[0:idx])
      root.right = self.buildTree(preorder[idx+1:], inorder[idx+1:])
      return root