题目描述:
不分行从上往下打印出二叉树的每个节点,同层节点从左至右打印。例如输入{8,6,10,#,#,2,1},如以下图中的示例二叉树,则依次打印8,6,10,2,1(空节点不打印,跳过),请你将打印的结果存放到一个数组里面,返回。
示例1
输入:{8,6,10,#,#,2,1}
返回值:[8,6,10,2,1]
思路:
1)采用二叉树的层次遍历法再结合队列的先进先出机制/辅助栈进行解题
对二叉树进行层次遍历,使用栈/队列存储遍历节点,并将该节点存入返回列表,判断该节点是否存在左右子树,若存在左右子树则将该节点的左右子树存入栈/队列,接下来取出栈/队列的第一个元素重复以上动作,直到栈/队列中的节点都存入到返回列表中,则输出返回列表
(2)使用存储节点的链表,控制链表为先进先出,按照题目要求的层序,设置Index存储节点读取的顺序,当链表连接一个新的节点时,将判断该节点是否有左右子节点,并按序添加到链表上
终止条件:由于最后读取的节点没有子节点,所以链表长度将不再增加,终止条件即为循环结束;终止后,遍历节点链表,将val赋值给int数组即可。
Python代码实现:
class Solution:
def PrintFromTopToBottom(self , root: TreeNode) -> List[int]:
# write code here
res=[]
if not root:
return []
# 将根节点入栈
stack=[root]
while stack:
# 获取二叉树每层的元素,搜索其左右子树
node=stack[0]
stack=stack[1:]
# 节点存入列表
res.append(node.val)
# 左节点入栈
if node.left:
stack.append(node.left)
# 右节点入栈
if node.right:
stack.append(node.right)
return res