【剑指 Offer 32 - III. 从上到下打印二叉树 III】

• ArrayList
• LinkedList

题目描述

请实现一个函数按照之字形顺序打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右到左的顺序打印，第三行再按照从左到右的顺序打印，其他行以此类推。
例如:
给定二叉树: [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

返回其层次遍历结果：

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

解题思路

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

思路1：设置标志flag

1. 设置一个标志flag，初始化为true；
2. 遍历奇数层
   • 当flag为true时（也就是树的第一（奇数）层），将队首元素添加至temp的尾部；
   • 遍历完一层后记得将true改为false；
3. 遍历偶数层
   • 当flag为false时（也就是树的第二（偶数）层），将队首元素添加至temp的头部；
   • 遍历完一层后记得将false改为true。

代码

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        Queue<TreeNode> queue = new LinkedList<>();
        List<List<Integer>> res = new ArrayList<>();
        if(root != null) queue.add(root);
        Boolean flag = true;

        while(!queue.isEmpty()){
            List<Integer> temp = new ArrayList<>();
            // LinkedList<Integer> temp = new LinkedList<>();
            for(int i = queue.size(); i > 0; i--){
                TreeNode head = queue.poll();
                if(flag == true) temp.add(head.val); // 默认在尾部添加元素
                else temp.add(0, head.val); // 在头部添加元素
                // if(flag == true) temp.addLast(head.val);
                // else temp.addFirst(head.val);

                if(head.left != null) queue.add(head.left);
                if(head.right != null) queue.add(head.right);
            }
            flag = (flag == true) ? false : true;
            res.add(temp);
        }
        return res;

    }
}

思路2：层序遍历 + 双端队列

判断最终返回的res（List<List>）大小是奇数还是偶数。因为每次都会将tmp降入到res中，所以res的层数就是树的层数。

• 奇数层 -> 队列尾部
• 偶数层 -> 队列头部

代码

Java 中将链表 LinkedList 作为双端队列使用。

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        Queue<TreeNode> queue = new LinkedList<>();
        List<List<Integer>> res = new ArrayList<>();
        if(root != null) queue.add(root);
        while(!queue.isEmpty()) {
            LinkedList<Integer> tmp = new LinkedList<>();
            for(int i = queue.size(); i > 0; i--) {
                TreeNode node = queue.poll();
                if(res.size() % 2 == 0) tmp.addLast(node.val); // 偶数层 -> 队列头部
                else tmp.addFirst(node.val); // 奇数层 -> 队列尾部
                if(node.left != null) queue.add(node.left);
                if(node.right != null) queue.add(node.right);
            }
            res.add(tmp);
        }
        return res;
    }
}

作者：jyd
链接：https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-iii-lcof/solution/mian-shi-ti-32-iii-cong-shang-dao-xia-da-yin-er--3/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

复杂度分析

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

知识点：ArrayList与LinkedList

ArrayList的创建和头尾添加元素。

 ArrayList<Integer> temp = new ArrayList<>();
 temp.add(head.val); // 在尾部添加元素
 temp.add(0, head.val); // 在头部添加元素

LinkedList的创建和头尾添加元素。

LinkedList<Integer> temp = new LinkedList<>();
temp.addLast(head.val);
temp.addFirst(head.val);