Description:
给定二叉树的层次及中序遍历,重建二叉树,输出该树的前序遍历及叶节点。
输入两行,分别为层次遍历及中序遍历。
输出两行,分为为叶节点及前序遍历。
Example:
Input:
3 5 4 2 6 7 1
2 5 3 6 4 7 1
Output:
2 6 1
3 5 2 4 6 7 1
思路: 对于子树,其在层次遍历最前面的点即为根节点,故重建过程包括以下两步:
1.利用层次遍历确定子树的根节点;
2.根据根节点在中序中的位置划定左右子树,递归重建。
代码:
class TreeNode: # 定义二叉树节点类
def __init__(self, x):
self.val = x
self.left = None
self.right = None
def restruct(level, mid):
if not level or not mid:
return None
lst = []
for num in mid:
lst.append(level.index(num))
idx = lst.index(min(lst))
root = TreeNode(mid[idx])
left = mid[0:idx]
right = mid[idx + 1:]
root.left = restruct(level, left)
root.right = restruct(level, right)
return root
lst1 = []
lst2 = []
def pre_traverse(root):
if not root:
return None
lst1.append(root.val)
pre_traverse(root.left)
pre_traverse(root.right)
return lst1
def leaf(root):
if not root:
return None
if not root.left and not root.right:
lst2.append(root.val)
leaf(root.left)
leaf(root.right)
return lst2
b = restruct([3, 5, 4, 2, 6, 7, 1], [2, 5, 3, 6, 4, 7, 1])
print(leaf(b))
print(pre_traverse(b))
输出:
类似:前序遍历+中序遍历重建二叉树:牛客OJ
思路: 重建过程包括以下递归的以下两步:
1.前序遍历的第一个节点为子树的根节点;
2.根据根节点在中序中的位置划定左右子树,递归重建。
代码:
class TreeNode: # 定义二叉树节点类
def __init__(self, x):
self.val = x
self.left = None
self.right = None
class Solution:
# 返回构造的TreeNode根节点
def reConstructBinaryTree(self, pre, tin):
if not pre or not tin:
return None
root=TreeNode(pre[0])
idx=tin.index(pre[0])
left=tin[0:idx]
right=tin[idx+1:]
root.left=self.reConstructBinaryTree(pre[1:len(left)+1],left)
root.right=self.reConstructBinaryTree(pre[-len(right):],right)
return root