从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。
思路:
使用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 []
queue = [root]
res=[]
while queue:
length = len(queue)
cur=[]
for _ in range(length):
node = queue.pop(0)
cur.append(node.val)
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
res.append(cur)
return res
来自:力扣leetcode