数据研发笔试Leetcode刷题笔记32：从上到下打印二叉树③

1 题目描述

来源：力扣（LeetCode）

请实现一个函数按照之字形顺序打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右到左的顺序打印，第三行再按照从左到右的顺序打印，其他行以此类推。

2 解题思路

面试题32 - I. 从上到下打印二叉树 主要考察 树的按层打印 ；
面试题32 - II. 从上到下打印二叉树 II 额外要求 每一层打印到一行 ；
本题额外要求 打印顺序交替变化（建议按顺序做此三道题）¹。

利用双端队列的两端皆可添加元素的特性，设打印列表（双端队列） tmp ，并规定：

• 奇数层 则添加至 tmp 尾部 ，
• 偶数层 则添加至 tmp 头部 。

算法流程：

• 特例处理： 当树的根节点为空，则直接返回空列表 [] ；
• 初始化： 打印结果空列表 res ，包含根节点的双端队列 deque ；
• BFS 循环： 当 deque 为空时跳出；
  • 新建列表 tmp ，用于临时存储当前层打印结果；
  • 当前层打印循环： 循环次数为当前层节点数（即 deque 长度）；
    • 出队： 队首元素出队，记为 node；
    • 打印： 若为奇数层，将 node.val 添加至 tmp 尾部；否则，添加至 tmp 头部；
    • 添加子节点： 若 node 的左（右）子节点不为空，则加入 deque ；
  • 将当前层结果 tmp 转化为 list 并添加入 res ；
• 返回值： 返回打印结果列表 res 即可；

3 代码实现（Python3）

class Solution:
    def levelOrder(self, root: TreeNode) -> List[List[int]]:
        if not root: return []
        res, deque = [], collections.deque()
        deque.append(root)
        while deque:
            tmp = collections.deque()
            for _ in range(len(deque)):
                node = deque.popleft()
                if len(res) % 2: tmp.appendleft(node.val) # 偶数层 -> 队列头部
                else: tmp.append(node.val) # 奇数层 -> 队列尾部
                if node.left: deque.append(node.left)
                if node.right: deque.append(node.right)
            res.append(list(tmp))
        return res

4 复杂度分析

• 时间复杂度 O(N) ： N 为二叉树的节点数量，即 BFS 需循环 N 次，占用 O(N) ；双端队列的队首和队尾的添加和删除操作的时间复杂度均为 O(1) 。
• 空间复杂度 O(N) ： 最差情况下，即当树为平衡二叉树时，最多有 N/2 个树节点同时在 deque 中，使用 O(N) 大小的额外空间。

深入了解复杂度：数据分析学习总结笔记11：空间复杂度和时间复杂度

---

1. 解题思路_作者：jyd ↩︎