标签: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