[剑指 Offer 32 - 从上到下打印二叉树 I ] | 刷题打卡

今天我们来做一道LeetCode上的题目，原题链接：剑指 Offer 32 - 从上到下打印二叉树 I

题目描述

• 从上到下打印出二叉树的每个节点，同一层的节点按照从左到右的顺序打印。
• 例如:
  • 给定二叉树: [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

• 返回：

  [3,9,20,15,7]

• 提示：

  节点总数 <= 1000

思路分析

• 二叉树遍历使用双端队列：collections.deque()

代码

# Python
# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def levelOrder(self, root):
        """
        :type root: TreeNode
        :rtype: List[int]
        """
        if not root: return []
        res, queue = [], collections.deque()
        queue.append(root)
        while queue:
            node = queue.popleft()
            res.append(node.val)
            if node.left: queue.append(node.left)
            if node.right: queue.append(node.right)
        return res

总结

• collections.deque()
  • append()：追加函数
  • popleft()：从头部取数据

附录

• leetcode-cn.com/problems/co…

本文正在参与「掘金 2021 春招闯关活动」, 点击查看 活动详情