题目描述:
给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
示例:
二叉树:[3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
个人思路:
这道题目明确要求使用层序遍历,按从左到右访问节点,根据输出提示来看,需要保存每一层的访问结果。我的思路就是:使用列表来保存每一层的非None的节点,然后利用该队列生成该层的访问结果,随后更新该列表为下一层的非None节点,本质思路也还是BFS。具体代码如下:
class Solution:
def levelOrder(self, root: TreeNode) -> List[List[int]]:
if not root:
return []
results = []
queue = []
queue.append(root)
while queue:
layer = [node.val for node in queue if node]
results.append(layer)
# 更新queue为下一层非None节点
temp = queue
queue = []
for node in temp:
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
return results
下面转载一位大神的BFS代码,以扩展思路:
class Solution(object):
def levelOrder(self, root):
queue = collections.deque()
queue.append(root)
res = []
while queue:
size = len(queue) # 通过设置for循环的长度,以达到记录每层访问结果的目的
level = []
for _ in range(size):
cur = queue.popleft()
if not cur:
continue
level.append(cur.val)
queue.append(cur.left)
queue.append(cur.right)
if level:
res.append(level)
return res
作者:fuxuemingzhu
链接:https://leetcode-cn.com/problems/binary-tree-level-order-traversal/solution/tao-mo-ban-bfs-he-dfs-du-ke-yi-jie-jue-by-fuxuemin/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。