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

"""
根据一棵树的前序遍历与中序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。

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

    3
   / \
  9  20
    /  \
   15   7
"""

def buildTree2(preorder, inorder):
    if len(inorder)==0:
        return None

    root = TreeNode(preorder[0])
    mid = inorder.index(preorder[0])

    root.left = buildTree2(preorder[1:mid+1], inorder[:mid])
    root.right = buildTree2(preorder[mid+1:], inorder[mid+1:])

    return root