【java】剑指offer32-III_从上到下打印二叉树III

题目描述

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

例如:
给定二叉树: [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7
返回其层次遍历结果：

[
  [3],
  [20,9],
  [15,7]
]
提示：

节点总数 <= 1000

我的解题思路：思路参考层序遍历二叉树打印II，再奇偶层不同时打印顺序不同，加一个字段flag判断是奇数层还是偶数层，如果是偶数则翻转节点集合后再追加进array中

public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> array = new ArrayList<>();
        Deque<TreeNode> deque = new LinkedList<>();
        if (root != null) {
           deque.offer(root);
        }
        boolean flag = false; // 从左至右
        while (!deque.isEmpty()) {
            List<Integer> list = new ArrayList<>(); // 每层节点存放集合
            for (int i = deque.size(); i > 0; i--) {
                TreeNode node = deque.poll(); // 取出节点
                list.add(node.val);
                if (node.left != null) {
                    deque.offer(node.left);
                }
                if (node.right != null) {
                    deque.offer(node.right);
                }
            }
            // 偶数行层数节点翻转加入节点集合
            if (flag) {
                Collections.reverse(list);
            }
            array.add(list);
            flag = !flag;
        }
        return array;
    }

 复杂度分析：
时间复杂度 O(N)： N为二叉树的节点数量，即需循环 N次，占用 O(N) 。
空间复杂度 O(N)： 最差情况下，即当树为满二叉树时，最多有N/2 个树节点 同时 在 deque 中，使用 O(N) 大小的额外空间。

 其他优秀思路：层序遍历 + 双端队列

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

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

算法流程：

• 特例处理： 当树的根节点为空，则直接返回空列表 [] ；
• 初始化： 打印结果空列表 res ，包含根节点的双端队列 deque ；
• BFS 循环： 当 deque 为空时跳出；

1. 新建列表 tmp ，用于临时存储当前层打印结果；
2. 当前层打印循环： 循环次数为当前层节点数（即 deque 长度）；
3. 出队： 队首元素出队，记为 node；
4. 打印： 若为奇数层，将 node.val 添加至 tmp 尾部；否则，添加至 tmp 头部；
5. 添加子节点： 若 node 的左（右）子节点不为空，则加入 deque ；
6. 将当前层结果 tmp 转化为 list 并添加入 res ；

• 返回值： 返回打印结果列表 res 即可；

public List<List<Integer>> levelOrder2(TreeNode root){
        List<List<Integer>> array = new ArrayList<>();
        Deque<TreeNode> deque = new LinkedList<>();
        if (root != null) {
            deque.offer(root);
        }
        while (!deque.isEmpty()) {
            LinkedList<Integer> list = new LinkedList<>(); // 双端队列每层节点存放集合
            for (int i = deque.size(); i > 0; i--) {
                TreeNode node = deque.poll();
                if (array.size() % 2 == 0) {
                    list.addLast(node.val); // 偶数层--放队尾
                } else {
                    list.addFirst(node.val); // 奇数层--放队头
                }
                if (node.left != null) {
                    deque.offer(node.left);
                }
                if (node.right != null) {
                    deque.offer(node.right);
                }
            }
            array.add(list);
        }
        return array;
    }

作者：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/