103. 二叉树的锯齿形层序遍历
问题103 二叉树的锯齿形层序遍历 ★★☆☆☆
https://leetcode-cn.com/problems/binary-tree-zigzag-level-order-traversal/
这个题目就是把层序遍历稍微改动下,如果从入队列时的顺序入手考虑,感觉就很麻烦,其实就还是按层序遍历入队出队,根据当前层级,加一个reverse就好了。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
if (root == null) {
return Collections.emptyList();
}
List<List<Integer>> ret = new ArrayList<>();
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
int curLevel = 0;
while (!queue.isEmpty()) {
int size = queue.size();
List<Integer> curLevelList = new ArrayList<>(size);
while (size-- > 0) {
TreeNode cur = queue.poll();
curLevelList.add(cur.val);
if (cur.left != null) {
queue.add(cur.left);
}
if (cur.right != null) {
queue.add(cur.right);
}
}
if (curLevel % 2 != 0) {
Collections.reverse(curLevelList);
}
ret.add(curLevelList);
curLevel++;
}
return ret;
}
本文由作者按照 CC BY 4.0 进行授权