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

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

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

例如，给出

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

    3
   / \
  9  20
    /  \
   15   7

链接：https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal

思路：
1、首先明确一个结论：根据二叉树的前序与中序遍历结果一定能还原二叉树
2、前序遍历的顺序是【根左右】，中序遍历的顺序是【左根右】。所以前序遍历的第一个节点是二叉树的根，对第一个节点在中序遍历中的索引进行分隔，左边是左子树，右边是右子树。递归上述过程，构造整个二叉树

Python代码：

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def buildTree(self, preorder, inorder):
        """
        :type preorder: List[int]
        :type inorder: List[int]
        :rtype: TreeNode
        """
        # 前序 1, 2,4,7, 3,5,6,8   根左右
        # 中序 4,7,2,1,5,3,8,6   左根右
        if not preorder:
            return None
        first = preorder.pop(0)
        ret = TreeNode(first)
        idx = inorder.index(first)
        ret.left = self.buildTree(preorder[:idx], inorder[:idx])
        ret.right = self.buildTree(preorder[idx:], inorder[idx+1:])

        return ret