剑指 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

关于树的遍历顺序

前序：根-》左子树-》右子树，其子树也符合前序遍历顺序。 中序：左子树-》根-》右子树，其子树也符合中序遍历顺序。

前序遍历的第一个节点一定是根节点，根据根节点给中序遍历来划分左右子树。 每次都能确定出三个节点的关系，重复这操作直至重构出二叉树。Krahets的题解里面的图文解析很好，我也是看完自己消化的，我不过是重复造轮子罢了。

# Definition for a binary tree node.
# 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:
    	'''
    	self.dic 是中序遍历中节点的索引，目的方便根据前序遍历中取出的值来划分中序中的左右子树
    	self.pre_order 是为了方便 recur 递归调用
    	'''
        self.dic, self.pre_order = {}, preorder
        for i in range(len(inorder)):
            self.dic[inorder[i]] = i
        return self.recur(0, 0, len(inorder)-1)

    def recur(self, pre_root, in_left, in_right):
    	'''
    	@pre_root: 前序中取出的节点的索引
    	                               ↓
    	@in_left: 中序遍历中左边界索引 [ 左 | 根 | 右 ]
    	                                          ↓
    	@in_right: 中序遍历中右边界索引 [ 左 | 根 | 右 ]
    	左右边界都是最左和最右的叶子
    	'''
    	# in_left 越过了 in_right 即子树中序遍历已空
        if in_left > in_right: return
        # 将前序遍历的节点当作根节点
        root = TreeNode(self.pre_order[pre_root])
        # 获得前序中的节点在中序中的索引，用于划分中序的左右子树
        i = self.dic[self.pre_order[pre_root]]
        # 左子树的根节点就是前序的根节点 + 1，右边界变成中序i - 1，这里一定要自己画图，或者去看看上面的题解
        root.left = self.recur(pre_root + 1, in_left, i - 1)
        # 同上
        root.right = self.recur(i - in_left + pre_root + 1, i + 1, in_right)
        return root