问题描述
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;
}
}