牛客网试题：python 重建二叉树代码注释

最新推荐文章于 2022-07-13 16:33:42 发布

原创最新推荐文章于 2022-07-13 16:33:42 发布 · 257 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种通过前序遍历和中序遍历结果重建二叉树的算法。核心内容包括定义Solution类，实现reConstructBinaryTree方法，通过递归方式构建二叉树结构。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

输入二叉树的前序遍历的前序遍历和中序遍历的结果，重建该二叉树

class Solution:
    # 返回构造的TreeNode根节点
    def reConstructBinaryTree(self, pre, tin):  #pre为前序遍历，tin为中序遍历
        # write code here                         
        if len(pre) == 0:     #如果前序遍历为空
            return None
        root = TreeNode(pre[0])   #根节点为前序遍历的第一个数字
        i = tin.index(pre[0])     # i为根节点在中序遍历中的位置
        root.left = self.reConstructBinaryTree(pre[1:1+i],tin[:i])  #获取左子树的前序遍历和中序遍历
        root.right = self.reConstructBinaryTree(pre[1+i:],tin[i+1:]) #获取右子树的前序遍历和中序遍历
        return root