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

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

题目描述

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

    3
   / \
  9  20
    /  \
   15   7

• 返回其层次遍历结果：

  [
    [3],
    [9,20],
    [15,7]
  ]

• 提示：

  节点总数 <= 1000

思路分析

• 二叉树遍历使用双端队列：collections.deque()
• 结果保存使用有序词典：collections.OrderedDict()

代码

# 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[List[int]]
        """
        if not root: return []
        res_dict, queue = collections.OrderedDict(), collections.deque()
        level = 0
        queue.append((root, level))
        while queue:
            node, level = queue.popleft()
            if level in res_dict:
                res_dict[level].append(node.val)
            else:
                res_dict[level] = [node.val]
            level += 1
            if node.left: queue.append((node.left, level))
            if node.right: queue.append((node.right, level))
        
        return res_dict.values()

总结

• collections.deque()
  • append()：追加函数
  • popleft()：从头部取数据
• collections.OrderedDict()
  • values()：返回值数组

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