码迷,mamicode.com
首页 > 编程语言 > 详细

利用层序遍历(不含None)和中序遍历重建二叉树 python

时间:2018-10-09 10:18:13      阅读:297      评论:0      收藏:0      [点我收藏+]

标签:return   二叉树的重建   class   形式   一个   序列   第一个   ini   one   

  给定一颗二叉树的层序遍历(不含None的形式)和中序遍历序列,利用两个序列完成对二叉树的重建。

  还是通过一个例子来说明整个过程,下图所示的二叉树,层序遍历结果为[a,b,c,d,e],中序遍历结果为[d,b,a,c,e],我们知道当我们找到根节点后,中序遍历能够提供给我们的信息就是左右子树分别包含哪些节点,而我们能否在层序遍历中找到根节点呢?很明显,层序遍历序列的第一个点是根节点,在中序遍历中找到左右子树分别包含哪些节点之后,我们就可以在层序遍历中区分出左右子树的层序遍历结果,然后就可以分别开始重建子树,明显这是一个重复的过程,所以我们用递归调用实现。

技术分享图片

  首先还是定义二叉树的类

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

  生成二叉树

    def buildtree(self,level,mid):
        if len(level)>0:
            root=TreeNode(level[0])#根节点
            rootid=mid.index(level[0])
            midleft=mid[:rootid]#左子树包含的节点
            midright=mid[rootid+1:]#右子树包含的节点
            levelleft,levelright=[],[]
            for node in level:
                if node in midleft:
                    levelleft.append(node)
                if node in midright:
                    levelright.append(node)
            root.left=self.buildtree(levelleft,midleft)
            root.right=self.buildtree(levelright,midright)
            return root

  Python版本:3.6

 

利用层序遍历(不含None)和中序遍历重建二叉树 python

标签:return   二叉树的重建   class   形式   一个   序列   第一个   ini   one   

原文地址:https://www.cnblogs.com/bambipai/p/9758652.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!