题目描述
从上往下打印出二叉树的每个节点,同层节点从左至右打印。
思路
先打印根节点,在遍历根的左右子树,再分别遍历左右子树的根节点,依次类推。
代码实现
# -*- coding:utf-8 -*-
# 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
ret=[]
if root==None:
return ret
bfs=[root]
while(bfs):
tbfs=[]
for node in bfs:
ret.append(node.val)
if node.left:
tbfs.append(node.left)
if node.right:
tbfs.append(node.right)
bfs=tbfs[:]
return ret
借助队列实现:
链接:https://www.nowcoder.com/questionTerminal/7fe2212963db4790b57431d9ed259701
来源:牛客网
class Solution:
# 返回从上到下每个节点值列表,例:[1,2,3]
def PrintFromTopToBottom(self, root):
# write code here
if not root:
return []
queue = []
result = [] #存储最后的遍历结果
queue.append(root) #根入队列
while len(queue) > 0:
node = queue.pop(0)
result.append(node.val)
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
return result