算法---LeetCode 103. 二叉树的锯齿形层序遍历(同剑指offer 32 - 3)

1. 题目

原题链接

给定一个二叉树，返回其节点值的锯齿形层序遍历。（即先从左往右，再从右往左进行下一层遍历，以此类推，层与层之间交替进行）。
例如：
给定二叉树 [3,9,20,null,null,15,7],

3

/
9 20
/
15 7

返回锯齿形层序遍历如下：

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

Related Topics 栈 树 广度优先搜索
👍 371 👎 0

2. 题解

2.1 解法1: 层序遍历, 维护标志变量, 在加入list 时判断顺序加入

    class Solution {
        public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
            List<List<Integer>> ans = new ArrayList<>();
            if (root == null) return ans;
            Queue<TreeNode> queue = new LinkedList<>();
            queue.offer(root);
            boolean flag = true;
            while (!queue.isEmpty()) {
                int size = queue.size();
                List<Integer> list = new ArrayList<>();
                for (int i = 0; i < size; i++) {
                    TreeNode temp = queue.poll();
                    if (flag) {
                        list.add(temp.val);
                    } else {
                        list.add(0, temp.val);
                    }
                    if (temp.left != null) {
                        queue.offer(temp.left);
                    }
                    if (temp.right != null) {
                        queue.offer(temp.right);
                    }
                }
                flag = !flag;
                ans.add(list);
            }
            return ans;
        }
    }

2.2 解法2: 层序遍历, 过程中根据奇偶逆序list

    class Solution {
        public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
            List<List<Integer>> listList = new ArrayList<>();
            if (root == null) {
                return listList;
            }
            Queue<TreeNode> queue = new LinkedList<>();
            queue.add(root);
            int flag = 1;
            while (!queue.isEmpty()) {
                int size = queue.size();
                ArrayList<Integer> list = new ArrayList<>();
                for (int i = 0; i < size; i++) {
                    TreeNode temp = queue.poll();
                    list.add(temp.val);
                    if (temp.left != null) {
                        queue.add(temp.left);
                    }
                    if (temp.right != null) {
                        queue.add(temp.right);
                    }
                }
                // 若为偶数层, 则翻转数组, 再加入结果集
                if (flag % 2 == 0) {
                    Collections.reverse(list);
                }
                listList.add(list);
                flag++;
            }
            return listList;
        }
    }

2.3 解法3 DFS

    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        List<List<Integer>> res = new ArrayList<>();
        travel(root, res, 0);
        return res;
    }

    private void travel(TreeNode cur, List<List<Integer>> res, int level) {
        if (cur == null)
            return;
        //如果res.size() <= level说明下一层的集合还没创建，所以要先创建下一层的集合
        if (res.size() <= level) {
            List<Integer> newLevel = new LinkedList<>();
            res.add(newLevel);
        }
        //遍历到第几层我们就操作第几层的数据
        List<Integer> list = res.get(level);
        //这里默认根节点是第0层，偶数层相当于从左往右遍历，
        // 所以要添加到集合的末尾，如果是奇数层相当于从右往左遍历，
        // 要把数据添加到集合的开头
        if (level % 2 == 0)
            list.add(cur.val);
        else
            list.add(0, cur.val);
        //分别遍历左右两个子节点，到下一层了，所以层数要加1
        travel(cur.left, res, level + 1);
        travel(cur.right, res, level + 1);
    }

参考: BFS和DFS两种解决方式