#4.重建2叉树
#输入二叉树的前序遍历和中序遍历的结果,重建出该二叉树
#前序遍历--根-->左-->右 中序遍历--左-->根-->右 后序遍历--左-->右-->根
#二叉树特例:①二叉搜索树--左子结点小于等于根结点,右子结点大于等于根结点②堆--分为最大堆和最小堆。在最大堆中根结点值最大,在最小堆中根结点值最小
class TreeNode:
def __init__(self,x):
self.val = x
self.left = None
self.right = None
# 前序遍历
pre = [1, 2, 4, 7, 3, 5, 6, 8]
#中序遍历
mid = [4, 7, 2, 1, 5, 3, 8, 6]
def findBinaryTree(pre,pre_start,pre_end,mid,mid_start,mid_end):
if pre_start>pre_end or mid_start>mid_end or len(pre)==0 or len(mid) == 0:
return None
#前序遍历第一个为根结点
root = TreeNode(pre[pre_start])
#循环遍历中序遍历,分别找到对应的左右子树
for i in range(mid_start,mid_end):
#在中序遍历中找到根结点--区分左右子树
if mid[i] == pre[pre_start]:
root.left = findBinaryTree(pre,pre_start+1,i-mid_start+pre_start,mid,mid_start,i-1)
root.right = findBinaryTree(pre,i-mid_start+pre_start+1,pre_end,mid,i+1,mid_end)
return root
print(findBinaryTree(pre,0,7,mid,0,7).val)
def findBinaryTree1(pre,nid):
if len(pre) == 0:
return None
if len(pre) == 1:
return TreeNode(pre[0])
else:
root = TreeNode(pre[0])
root.left = findBinaryTree1(pre[1:mid.index(pre[0])+1],mid[:mid.index(pre[0])])
root.right = findBinaryTree1(pre[mid.index(pre[0])+1:],mid[mid.index(pre[0])+1:])
return root
重建二叉树--python
最新推荐文章于 2021-08-24 19:25:14 发布