【力扣算法】103-二叉树的锯齿形层次遍历

题目

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

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

    3
   / \
  9  20
    /  \
   15   7

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/binary-tree-zigzag-level-order-traversal
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题解

无官方题解

感想

“新三年，旧三年，缝缝补补又三年”，100、101、102的代码改来改去用到现在了。这道题的思路很简单，就是在102题目的基础上，要么直接把res的偶数行的LinkedList都颠倒一下就行了；要么就是利用LinkedList既实现了Queue接口，又实现了Deque接口的性质，把LinkedList当双端队列用，轮流使用不同方向读取队列。

自己实现使用的是第二套方案，感觉时间消耗更少些：

执行用时 :2 ms, 在所有 Java 提交中击败了94.50%的用户

内存消耗 :35.3 MB, 在所有 Java 提交中击败了44.22%的用户

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        List<List<Integer>> res = new LinkedList<>();
        if(root==null) return res;
        LinkedList<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        TreeNode nextFirst = null;
        TreeNode temp = null;
        boolean leftToRight = true;
        List<Integer> list = new LinkedList<>();
        while(!queue.isEmpty()){
            if(leftToRight) temp = queue.peek();
            else temp = queue.peekLast();
            
            if(temp==nextFirst) {
                nextFirst=null;
                res.add(list);
                list = new LinkedList<>();
                leftToRight=!leftToRight;
                continue;
            }
            list.add(temp.val);
            if(leftToRight){
                if(temp.left!=null){
                    queue.add(temp.left);
                    if(nextFirst==null) nextFirst = temp.left;
                }
                if(temp.right!=null){
                    queue.add(temp.right);
                    if(nextFirst==null) nextFirst = temp.right;
                }
                queue.poll();
            }
            else{
                if(temp.right!=null){
                    queue.addFirst(temp.right);
                    if(nextFirst==null) nextFirst = temp.right;
                }
                if(temp.left!=null){
                    queue.addFirst(temp.left);
                    if(nextFirst==null) nextFirst = temp.left;
                }
                queue.pollLast();
            }
        }
        if(list.size()>0) res.add(list);
        return res;
    }
}

后来参考别人的答案意识到一个问题：如果LinkedList仍然当做Queue始终在一端poll和add，而根据leftToRight只改输出的list的add的方向的话，就不需要双端队列了，可以稍微简化代码。

就是把while循环内改成：

temp = queue.poll();
if(temp==nextFirst) {
    nextFirst=null;
    res.add(list);
    list = new LinkedList<>();
    leftToRight=!leftToRight;
    continue;
}
if(LeftToRight) list.add(temp.val);
else list.addFirst(temp.val);
if(temp.left!=null){
    queue.add(temp.left);
    if(nextFirst==null) nextFirst = temp.left;
}
if(temp.right!=null){
    queue.add(temp.right);
    if(nextFirst==null) nextFirst = temp.right;
}