剑指 Offer 07. 重建二叉树 python

题目

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

例如，给出

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

3

/
9 20
/
15 7

限制：

0 <= 节点个数 <= 5000

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/zhong-jian-er-cha-shu-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

很经典的问题

DLR–前序遍历（根在前，从左往右，一棵树的根永远在左子树前面，左子树又永远在右子树前面 ）
LDR–中序遍历（根在中，从左往右，一棵树的左子树永远在根前面，根永远在右子树前面）
LRD–后序遍历（根在后，从左往右，一棵树的左子树永远在右子树前面，右子树永远在根前面）

前序遍历 根在最前面
中序遍历 根在中间
后序遍历 根在最后

代码

class Solution(object):
    def buildTree(self, preorder, inorder):
        """
        :type preorder: List[int]
        :type inorder: List[int]
        :rtype: TreeNode
        """

        if len(preorder):
            root = preorder[0]

            tree_node = TreeNode(root)

            if root in inorder:
                r_index = inorder.index(root)
                left, right = inorder[:r_index], inorder[r_index + 1:]
                if len(left):
                    tree_node.left = self.buildTree(preorder[1:], left)
                if len(right):
                    tree_node.right = self.buildTree(preorder[len(left)+1:], right)
            return tree_node
        else:
            return