面试题22:从上往下打印二叉树
2018-07-29 本文已影响0人
fighting_css
【题目描述】
从上往下打印出二叉树的每个节点,同层节点从左至右打印。
【思路】
新加入一个节点时,将该节点的左右子树添加进队列
image.png
image.png
【代码】:
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
# 返回从上到下每个节点值列表,例:[1,2,3]
def PrintFromTopToBottom(self, root):
# write code here
if root==None:
return []
res = []
queue = [root]
#将根加入队列
while queue:
#取出队列的头结点
pNode_list = queue
queue = []
for pNode in pNode_list:
#结果队列加入相应的节点值
res.append(pNode.val)
if pNode.left:
queue.append(pNode.left)
if pNode.right:
queue.append(pNode.right)
#将头节点取出后,更新队列
return res