本文共 1583 字，大约阅读时间需要 5 分钟。

题目难度: 中等

今天继续更新剑指 offer 系列, 这道题和接下来两道题很类似, 属于非常经典的 BFS, 这里提供两种实现方案供大家参考

老样子晚上 6 点 45 分准时更新公众号 每日精选算法题, 大家记得关注哦~ 另外在公众号里回复 offer 就能看到剑指 offer 系列当前连载的所有文章了

题目描述

从上到下打印出二叉树的每个节点，同一层的节点按照从左到右的顺序打印。

• 节点总数 <= 1000

题目样例

示例

输入

给定二叉树: [3,9,20,null,null,15,7],

3   / \  9  20    /  \   15   7

输出

[3,9,20,15,7]

题目思考

1. 哪些数据结构可以做到按层输出?

解决方案

思路

• 经典的 BFS 思路, 将当前节点的左右非空子节点追加到队列结尾, 然后继续往后遍历队列, 最终整个队列就是最后的结果
• 由于这里是树, 所以每个节点只可能被加入队列访问一次, 无需额外的 visit 集合
• 另外这道题不需要分开保存每一层的节点, 所以这里不需要记录当前层的长度之类的情况, 可以直接一次循环搞定
• 数据结构方面, 可以采用列表 + 动态的 for 循环(python 3 适用, 其他语言可能不支持遍历过程中更改容器)或者双端队列 + while 循环(所有语言通用, 每次 pop 最左边的元素作为当前节点)
• 下面的代码是个人觉得比较通用的 BFS 模板, 包括列表和双端队列两种方式的实现, 适用于不需要单独处理每一层节点的情况, 供大家参考

复杂度

• 时间复杂度 O(N)
  • 每个节点只需要遍历一次
• 空间复杂度 O(N)
  • 额外需要一个列表/双端队列

代码

方案 1 - 列表 + for 循环

class Solution:    def levelOrder(self, root: TreeNode) -> List[int]:        # 方案1: 列表+for循环        if not root:            return []        q = [root]        for node in q:            # 只将非空子节点追加到队列            if node.left:                q.append(node.left)            if node.right:                q.append(node.right)        return [x.val for x in q]

方案 2 - 双端队列 + while 循环

import collectionsclass Solution:    def levelOrder(self, root: TreeNode) -> List[int]:        # 方案2: 双端队列+while循环        if not root:            return []        q = collections.deque([root])        res = []        while q:            node = q.popleft()            res.append(node.val)            if node.left:                q.append(node.left)            if node.right:                q.append(node.right)        return res

大家可以在下面这些地方找到我~😊

我的公众号: 每日精选算法题, 欢迎大家扫码关注~😊

转载地址：https://blog.csdn.net/zjulyx1993/article/details/107367942 如侵犯您的版权，请留言回复原文章的地址，我们会给您删除此文章，给您带来不便请您谅解！