LeetCode103. 二叉树的锯齿形层次遍历

题目

给定一个二叉树，返回其节点值的锯齿形层次遍历。（即先从左往右，再从右往左进行下一层遍历，以此类推，层与层之间交替进行）。

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

    3
   / \
  9  20
    /  \
   15   7

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

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

解题

• 二叉树层序遍历的变种
• 使用队列辅助解题
• 需要一个额外的变量记录遍历顺序, 代码如下:

class Solution {
    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        if (root == null) {
            return Collections.emptyList();
        }
        LinkedList<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        List<List<Integer>> result = new ArrayList<>();
        // 用于记录层序遍历顺序, 从左到右还是从右到左
        boolean isLeftToRight = true;
        while (!queue.isEmpty()) {
            int size = queue.size();
            List<Integer> levelList = new ArrayList<>();
            for (int i = 0; i < size; i++) {
                TreeNode node;
                // 当前层从左到右遍历
                if (isLeftToRight) {
                    node = queue.poll();
                    levelList.add(node.val);
                    if (node.left != null) {
                        queue.add(node.left);
                    }
                    if (node.right != null) {
                        queue.add(node.right);
                    }
                    // 当前层从右往左遍历
                } else {
                    node = queue.removeLast();
                    levelList.add(node.val);
                    // 下一层从左往右遍历, 所以需要从前面开始添加元素,而且从右向左
                    if (node.right != null) {
                        queue.addFirst(node.right);
                    }
                    if (node.left != null) {
                        queue.addFirst(node.left);
                    }
                }

            }
            // 修改遍历顺序
            isLeftToRight = !isLeftToRight;
            result.add(levelList);
        }

        return result;
    }

}