剑指offer笔记4：重建二叉树（Python）

剑指offer笔记4：重建二叉树（P62）

题目：重建二叉树

输入某二叉树的前序遍历和中序遍历的结果，请重建二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建二叉树并返回头结点。二叉树节点的定义如下：

class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

问题分析

以下图的二叉树为例

该二叉树的先序遍历序列为{1,2,4,7,3,5,6,8}，中序遍历序列为{4,7,2,1,5,3,8,6}。
先序遍历的顺序是根节点→左子节点→右子节点
中序遍历的顺序是左子节点→根节点→右子节点
因此，在先序遍历序列中，第一个元素即为二叉树的根。对应到中序遍历序列中，根元素所在位置以左的为其左子树的中序遍历序列，根元素所在位置以右的为其右子树的中序遍历序列。如下图所示

我们可以通过以下步骤，递归地构造二叉树：

1. 先序遍历序列第一个元素为根
2. 找到根元素在中序遍历序列中的位置
3. 计算出先序和中序遍历序列中左子树和右子树遍历序列的范围
4. 递归的计算左右子树

python代码

    def reConstructBinaryTree(self, pre, tin):
        if len(pre) == 0:
            # 若二叉树为空则返回None
            return None
        if len(pre) == 1:
            # 若二叉树只有一个节点，则返回该节点
            return TreeNode(pre[0])
        root = TreeNode(pre[0])
        # 根为先序遍历序列的第一个元素
        mid = tin.index(pre[0])
        # 用mid定位中序遍历序列中根元素的位置
        root.left = self.reConstructBinaryTree(pre[1:mid+1], tin[:mid])
        # 递归的构建左子树
        root.right = self.reConstructBinaryTree(pre[mid+1:], tin[mid+1:])
        # 递归的构建右子树
        return root

小结

1. 已知先序遍历中序遍历和已知中序遍历后序遍历，构造二叉树的方法类似
2. 有关二叉树的很多问题都可通过递归来解决

附：已知中序遍历和后序遍历，构造二叉树的代码

    def reConstruct_post_in(self, post, tin):
        if len(post) == 0:
            # 若二叉树为空则返回None
            return None
        if len(post) == 1:
            # 若二叉树只有一个节点，则返回该节点
            return TreeNode(post[0])
        root = TreeNode(post[-1])
        # 根为后序遍历序列的最后一个元素
        mid = tin.index(post[-1])
        # 用mid定位中序遍历序列中根元素的位置
        root.left = self.reConstruct_post_in(post[:mid], tin[:mid])
        # 递归的构建左子树
        root.right = self.reConstruct_post_in(post[mid:-1], tin[mid+1:])
        # 递归的构建右子树
        return root