LeetCode levelOrder Python3

最新推荐文章于 2022-05-07 21:36:19 发布

Whisper321

最新推荐文章于 2022-05-07 21:36:19 发布

阅读量172

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/Whisper321/article/details/96021562

版权

LeetCode 专栏收录该内容

24 篇文章 0 订阅

订阅专栏

题目描述：给定一颗二叉树，返回其按层次遍历的节点值（即逐层地，从左到右访问所有节点）

示例：
输入：[3,9,20,null,null,15,7]
输出：
[
[3],
[9,20],
[15,7]
]

这道题很容易想到使用 BFS 按层遍历二叉树的所有节点，比较困难的点在于输出的形式：需要将同层的节点组合在一起，所以使用一个临时队列 temp_q 记录下一层的节点，当 queue 中的节点全部出队时，即遍历完了这一层的所有节点，此时再将临时队列赋给 queue，临时队列置空。

def levelOrder(root):
	if not root:
        return []
    queue = [root]
    temp_q, temp = [], []
    res = []
    while len(queue)>0:
        cur = queue.pop(0)
        if cur.left:
            temp_q.append(cur.left)
        if cur.right:
            temp_q.append(cur.right)
        temp.append(cur.val)
        if len(queue) == 0:
            res.append(temp)
            queue = temp_q
            temp_q, temp = [], []
    return res