剑指offer 7 - 重建二叉树 - python

题目描述：

输入某二叉树的前序遍历和中序遍历的结果，请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

例如，给出

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树：

    3
   / \
  9  20
    /  \
   15   7
 

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

根节点preorder[0]在中序遍历数列中的位置index，index左边的部分为左子树的中序遍历序列，index右边的为右子树的中序遍历序列。相应的在前序遍历中，从索引1开始后的部分为对应的左子树和右子树的前序遍历序列，它们的元素个数和中序遍历中划分后的个数相同。 然后将左子树和右子树分别当作一棵新的二叉树，递归调用函数进行重建。

`

# -*- coding:utf-8 -*-
# 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 preorder == [] or inorder == []:
            return None

        root = TreeNode(preorder[0])
        index = inorder.index(preorder[0])
        root.left = self.buildTree(preorder[1:index + 1], inorder[:index])
        root.right = self.buildTree(preorder[index + 1:], inorder[index + 1:])

        return root