题目
从上到下打印二叉树
一、思路
从上到下打印二叉树其实就是树的层序遍历~也叫广度搜索~
我们可以虚拟一个队列,每次都把遍历到的当前节点的子节点(下一层)放入队尾
def print_top_to_bottom_tree(tree):
if not tree:
return []
res = []
res.append(tree.root)
print_list = []
if tree.left:
print_list.append(tree.left)
if tree.right:
print_list.append(tree.right)
for node in print_list:
res.append(node.root)
if node.left:
print_list.append(node.left)
if node.right:
print_list.append(node.right)
return res