跳至主要內容

剑指 Offer 32 - I. 从上到下打印二叉树

ITCharge小于 1 分钟

剑指 Offer 32 - I. 从上到下打印二叉树open in new window

  • 标签:树、广度优先搜索、二叉树
  • 难度:中等

题目链接

题目大意

给定一棵二叉树的根节点 root

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

解题思路

广度优先搜索。

具体步骤如下:

  • 根节点入队。
  • 当队列不为空时,求出当前队列长度 sis_i
    • 依次从队列中取出这 sis_i 个元素,将其加入答案数组,并将其左右子节点入队,然后继续迭代。
  • 当队列为空时,结束。

代码

class Solution:
    def levelOrder(self, root: TreeNode) -> List[int]:
        if not root:
            return []
        queue = [root]
        order = []
        while queue:
            size = len(queue)
            for _ in range(size):
                curr = queue.pop(0)
                order.append(curr.val)
                if curr.left:
                    queue.append(curr.left)
                if curr.right:
                    queue.append(curr.right)
        return order