题目描述
从上往下打印出二叉树的每个节点,同层节点从左至右打印。
思路
二叉树层次遍历,用队列存储每层结点,再依次弹出。
def PrintFromTopToBottom(root):
# write code here
if not root: return []
queue = [root]
res = []
while queue:
n = len(queue)
for _ in range(n):
if not queue: break
node = queue.pop(0)
res.append(node.val)
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
return res