题目描述
从上往下打印出二叉树的每个节点,同层节点从左至右打印。
思路
从上到下打印二叉树的规律:
每次打印一个节点后,如果该节点有子结点,则把该节点的子节点放到一个队列的末尾。接下来到队列的头部取出最早进入队列的节点,重复前面的打印操作,知直到队列的所有节点都被打印出来。
代码实现
class TreeNode:
def __init__(self, x):
self.val = x
self.left = None
self.right = None
class Solution:
# 返回从上到下每个节点值列表,例:[1,2,3]
def PrintFromTopToBottom(self, root):
l=[]
if root==None:
return None
q=[root]
while len(q):
t=q.pop(0)
l.append(t.val)
if t.left:
q.append(t.left)
if t.right:
q.append(t.right)
return l
#测试
node1=TreeNode(8)
node2=TreeNode(6)
node3=TreeNode(10)
node4=TreeNode(5)
node5=TreeNode(7)
node6=TreeNode(9)
node7=TreeNode(11)
node1.left=node2
node1.right=node3
node2.left=node4
node2.right=node5
node3.left=node6
node3.right=node7
S=Solution()
temp=S.PrintFromTopToBottom(node1)
print(temp)