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

问题描述

LeetCode 103. 二叉树的锯齿形层序遍历 (opens in a new tab),难度中等

给你二叉树的根节点 root ,返回其节点值的 锯齿形层序遍历 。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。

示例 1

输入:root = [3,9,20,null,null,15,7]
输出:[[3],[20,9],[15,7]]

示例 2

输入:root = [1]
输出:[[1]]

示例 3

输入:root = []
输出:[]

提示:

  • 树中节点数目在范围 [0, 2000]
  • -100 <= Node.val <= 100

题解

题解:这道题开始想麻烦了,第一眼思路是“之”字型存储 TreeNode 到 Queue。其实只需要将结果按“之”字型输出。

Solution.java
class Solution {
    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        List<List<Integer>> ans = new ArrayList<>();
        if (root == null) return ans;
 
        Queue<TreeNode> queue = new ArrayDeque<>();
        queue.add(root);
        // 用于标识当前的层级
        int level = 1;
        while (!queue.isEmpty()) {
            List<Integer> temp = new ArrayList<>();
            int size = queue.size();
            for (int i = 0; i < size; ++i) {
                TreeNode tn = queue.poll();
                temp.add(tn.val);
                if (tn.left != null) {
                    queue.offer(tn.left);
                }
                if (tn.right != null) {
                    queue.offer(tn.right);
                }
            }
            // 偶数层级,反转顺序
            if (level % 2 == 0) Collections.reverse(temp);
            level++;
            ans.add(temp);
        }
        return ans;
    }
}