题目链接
题目描述
给你二叉树的根节点 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
解题思路
层序遍历,然后奇数层顺序不变,偶数层反转数组
AC代码
class Solution {
public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
List<List<Integer>> ans = new ArrayList<>();
if (root == null) {
return ans;
}
Deque<TreeNode> queue = new LinkedList<>();
queue.offer(root);
boolean flag = true;
while (!queue.isEmpty()) {
int len = queue.size();
List<Integer> itemList = new ArrayList<>();
for (int i = 0; i < len; i++) {
TreeNode tmp = queue.poll();
itemList.add(tmp.val);
if (tmp.left != null) {
queue.offer(tmp.left);
}
if (tmp.right != null) {
queue.offer(tmp.right);
}
}
if (flag) {
ans.add(itemList);
} else {
Collections.reverse(itemList);
ans.add(itemList);
}
flag = !flag;
}
return ans;
}
}