Sword32-III——从上到下打印二叉树 III

Sword32-III——从上到下打印二叉树 III

方法1——BFS

• 思路：此题也称为之字形遍历，与上一题的解法类似。只是需要判断当前是奇数偶数层，奇数层则tmp可不变，偶数层则tmp需逆序（或者使用双端队列从头部弹出），此处采取双端队列tmp
  • 如何判断是奇数层还是偶数层？
    • 通过res的size进行判断，当res的size大小为偶数，说明正在遍历的就是奇数层；反之为奇数，说明正在遍历的为偶数层
• 特殊情况与临界分析：根节点为空
• 终止条件：队列中不存在元素
• 步骤：
  • 定义一个存储结果res的ArrayList，便于校验中直接返回空ArrayList
  • 根节点为空校验
  • 定义一个辅助队列
  • 定义一个移动节点
  • 将移动节点（即根节点）入队
  • 循环条件：队列非空
  • while循环
    • 定义一个**双端队列（LinkedList充当）**用于存储每一层结果
    • for循环，因为queue的size存入元素后将会变化，因此需要记录当前queue的size大小，因此将i直接记录queue.size()，使得i依次减1即可，完成之后即弹出上一层的所有节点
      • 从队列中弹出一个元素
      • 判断res的size大小是奇数还是偶数
        • 结果为偶数，说明当前为奇数层，将其加入到tmp的尾部
        • 结果为奇数，说明当前为偶数层，将其加入到tmp的首部
      • 判断该节点左子节点是否非空，非空则入队
      • 判断该节点右子节点是否非空，非空则入队
    • 此时tmp已经装满了上一层的结果，将其加入到res中
  • 返回结果（此处不需要转换为数组形式）

    public List<List<Integer>> levelOrder(TreeNode root) {
        // 定义存储结果的ArrayList
        List<List<Integer>> res = new ArrayList<>();
        // 根节点非空校验
        if (root == null) {
            return res;
        }
        // 定义辅助队列
        Queue<TreeNode> queue = new LinkedList<>();
        // 定义移动节点
        TreeNode cur = root;
        // 将移动节点入队
        queue.offer(cur);
        // while循环
        while (!queue.isEmpty()) {
            // 双端队列
            LinkedList<Integer> tmp = new LinkedList<>();
            // for循环
            for (int i = queue.size(); i > 0; i--) {
                // 弹出一个节点
                cur = queue.poll();
                // 判断当前层是奇数层还是偶数层
                // 此处为偶数，说明当前层为奇数层，加入tmp尾部
                if ((res.size() & 1) == 0) {
                    tmp.addLast(cur.val);
                // 此处为奇数，说明当前层为偶数层，加入tmp首部
                } else {
                    tmp.addFirst(cur.val);
                }
                // 判断其左子节点是否非空
                if (cur.left != null) {
                    // 非空则将左子节点入队
                    queue.offer(cur.left);
                }
                // 判断其右子节点是否非空
                if (cur.right != null) {
                    // 非空则将右子节点入队
                    queue.offer(cur.right);
                }
            }
            // for循环完毕，将tmp加入res中
            res.add(tmp);
        }
        // 返回结果
        return res;
    }