【Leetcode】103. 二叉树的锯齿形层序遍历

题目描述

// 103. 二叉树的锯齿形层序遍历


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

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

题解

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
// 本题和【剑指offer】32.3 从上到下打印二叉树3 / 按之字形顺序打印二叉树 
// 一模一样。
// 想不出来的可以先写【剑指offer】32.1 从上到下打印二叉树 
// 再写 【剑指offer】32.2 从上到下打印二叉树2 
// 再写出最终的解答。
// 这里使用reverse 布尔值控制层的子数组是否翻转
// 
// 执行用时：1 ms, 在所有 Java 提交中击败了98.67%的用户
// 内存消耗：38.6 MB, 在所有 Java 提交中击败了55.38%的用户
import java.util.LinkedList;
class Solution {
    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        List<List<Integer>> res = new ArrayList<>();
        if (root == null)
            return res;
        Queue<TreeNode> q = new LinkedList<>();
        boolean reverse = false;
        q.add(root);
        while (!q.isEmpty()) {
            int len = q.size();
            List<Integer> level = new ArrayList<>();
            while (len > 0) {
                TreeNode node = q.remove();
                level.add(node.val);
                if (node.left != null)
                    q.add(node.left);
                if (node.right != null)
                    q.add(node.right);
                len--;
            }
            if (reverse) {
                Collections.reverse(level);
            }
            reverse = !reverse;
            res.add(level);
        }
        return res;
    }
}



// 优化一下
// 这里用reverse 布尔值控制向level中插入元素的方向。
// 
// 执行用时：1 ms, 在所有 Java 提交中击败了98.67%的用户
// 内存消耗：38.3 MB, 在所有 Java 提交中击败了95.39%的用户
import java.util.LinkedList;
class Solution {
    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        List<List<Integer>> res = new ArrayList<>();
        if (root == null)
            return res;
        Queue<TreeNode> q = new LinkedList<>();
        boolean reverse = false;
        q.add(root);
        while (!q.isEmpty()) {
            int len = q.size();
            List<Integer> level = new ArrayList<>();
            while (len > 0) {
                TreeNode node = q.remove();
                if (reverse) 
                    level.add(0, node.val);
                else 
                    level.add(node.val);
                if (node.left != null)
                    q.add(node.left);
                if (node.right != null)
                    q.add(node.right);
                len--;
            }
            reverse = !reverse;
            res.add(level);
        }
        return res;
    }
}