从上到下打印出二叉树的每个节点,同一层的节点按照从左到右的顺序打印。
例如:
给定二叉树: [3,9,20,null,null,15,7]
,
3
/ \
9 20
/ \
15 7
返回:
[3,9,20,15,7]
常规的层次遍历。
class Solution:
def levelOrder(self, root: TreeNode) -> List[int]:
result = []
if root:
stack = [root]
while stack:
node = stack.pop(0)
if node.left is not None:
stack.append(node.left)
if node.right is not None:
stack.append(node.right)
result.append(node.val)
return result
解题思路:
- 题目要求的二叉树的 从上至下 打印(即按层打印),又称为二叉树的 广度优先搜索(BFS)。
- BFS 通常借助 队列 的先入先出特性来实现。
算法流程:
- 特例处理: 当树的根节点为空,则直接返回空列表 [] ;
- 初始化: 打印结果列表
res = []
,包含根节点的队列queue = [root]
; - BFS 循环: 当队列
queue
为空时跳出;- 出队: 队首元素出队,记为
node
; - 打印: 将
node.val
添加至列表res
尾部; - 添加子节点: 若
node
的左(右)子节点不为空,则将左(右)子节点加入队列queue
;
- 出队: 队首元素出队,记为
- 返回值: 返回打印结果列表
res
即可
class Solution:
def levelOrder(self, root: TreeNode) -> List[int]:
if not root: return []
res, queue = [], collections.deque()
queue.append(root)
while queue:
node = queue.popleft()
res.append(node.val)
if node.left: queue.append(node.left)
if node.right: queue.append(node.right)
return res