从上到下打印出二叉树的每个节点,同一层的节点按照从左到右的顺序打印。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回:
[3,9,20,15,7]
我们从根节点开始按宽度优先的顺序遍历整棵树,每次先扩展左儿子,再扩展右儿子 (BFS) 。
先扩展根节点;
再依次扩展根节点的左右儿子,也就是从左到右扩展第二层节点;
再依次从左到右扩展第三层节点;
依次类推,所以BFS的顺序就是这道题目要求的顺序。
时间复杂度
BFS时每个节点仅被遍历一次,所以时间复杂度是 O(n)。
# 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[int]:
if not root:
return []
res = []
q = [root]
while q:
node = q.pop(0)
res.append(node.val)
if node.left:
q.append(node.left)
if node.right:
q.append(node.right)
return res
来源:力扣(LeetCode)