剑指offer 重建二叉树 python

题目描述

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

样例

输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}
返回二叉树头节点

---

想法：
使用递归，既然给你前序遍历序列和中序遍历序列，所以可以通过前序遍历的第一个为中间节点，每次都划分出根节点左子树和右子树的前序遍历序列和中序遍历序列传给函数递归，最后返回根节点。

class Solution:
    # 返回构造的TreeNode根节点
    def reConstructBinaryTree(self, pre, tin):
        """
        :param pre: List
        :param tin: List
        :return: TreeNode
        """
        if len(pre) == 0:
            return None
        if len(pre) == 1:
            return TreeNode(pre[0])

        # 中左 中右
        zz, zy = tin[:tin.index(pre[0])], tin[tin.index(pre[0]) + 1:]

        # 前左 前右
        qz, qy = pre[1:len(zz)+1], pre[len(zz)+1:]

        res = TreeNode(pre[0])
        res.left = Solution().reConstructBinaryTree(qz, zz)
        res.right = Solution().reConstructBinaryTree(qy, zy)

---

最后

刷过的LeetCode源码放在Github上了，希望喜欢或者觉得有用的朋友点个star或者follow。
有任何问题可以在下面评论或者通过私信或联系方式找我。
联系方式
QQ：791034063
Wechat：liuyuhang791034063
CSDN：https://blog.csdn.net/Sun_White_Boy
Github：https://github.com/liuyuhang791034063