从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。
思路:
使用BFS的思路进行存储二叉树中的节点,在打印,其中使用了队列的性质
首先将第一个结点放入队列q中,如果节点不为空,很容易的知道第一层的输出就是这个节点。把这个节点存入到第一层的结果中,接着遍历当前的左右节点,若果不是空的话就把它加入到临时数组out中,然后更新q = out,这时的q就是下一层的节点,以此类推,直到q中没有节点,说明遍历完成。
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def levelOrder(self, root: TreeNode) -> List[List[int]]:
if not root:
return []
res = []
q = [root]
while q:
out = []
temp = []
for i in q:
temp.append(i.val)
if i.left:
out.append(i.left)
if i.right:
out.append(i.right)
res.append(temp)
q = out
return res
来自:力扣leetcode