剑指offer 第二版（Python3）--面试题7：重建二叉树

第2章 面试需要的基础知识

  面试题4：二维数组中的查找

  面试题5：替换空格

  面试题6：从尾到头打印链表

  面试题7：重建二叉树

第3章 高质量的代码

第4章 解决面试题的思路

第5章 优化时间和空间效率

第6章 面试中的各项能力

第7章 两个面试案例

题目描述
来源：牛客网
  输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建二叉树并返回。

解题思路
来源：牛客网
  根据中序遍历和前序遍历可以确定二叉树，具体过程为：

1. 根据前序序列第一个结点确定根结点
2. 根据根结点在中序序列中的位置分割出左右两个子序列
3. 对左子树和右子树分别递归使用同样的方法继续分解

例如：
    前序序列{1,2,4,7,3,5,6,8} = pre
    中序序列{4,7,2,1,5,3,8,6} = tin

1. 根据当前前序序列的第一个结点确定根结点，为 1
2. 找到 1 在中序遍历序列中的位置，为tin[3]
3. 切割左右子树，则 tin[3] 前面的为左子树， tin[3] 后面的为右子树
4. 则切割后的左子树前序序列为：{2,4,7}，切割后的左子树中序序列为：{4,7,2}；切割后的右子树前序序列为：{3,5,6,8}，切割后的右子树中序序列为：{5,3,8,6}
5. 对子树分别使用同样的方法分解

实战

# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    # 返回构造的TreeNode根节点
    def reConstructBinaryTree(self, pre, tin):
        # write code here
        if not pre:
            return

        index = tin.index(pre[0])
        node = TreeNode(tin[index])

        node.left = self.reConstructBinaryTree(pre[1:index + 1], tin[:index])
        node.right = self.reConstructBinaryTree(pre[index + 1:], tin[index + 1:])

        return node