-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathLC_103.java
55 lines (49 loc) · 1.52 KB
/
LC_103.java
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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
import java.util.ArrayList;
import java.util.Deque;
import java.util.LinkedList;
import java.util.List;
public class LC_103 {
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;
}
}
public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
Deque<TreeNode> que = new LinkedList<>();
List<List<Integer>> res = new ArrayList<>();
if (root == null) return res;
que.addLast(root);
int level = 0;
while (!que.isEmpty()) {
int size = que.size();
List<Integer> path = new ArrayList<>();
while (size-- > 0) {
TreeNode node = que.pollFirst();
if (node != null) {
path.add(node.val);
if (node.left != null) que.addLast(node.left);
if (node.right != null) que.addLast(node.right);
}
}
if (level++ % 2 == 0) res.add(new ArrayList<>(path));
else res.add(reverse(path));
}
return res;
}
public static List<Integer> reverse(List<Integer> path) {
List<Integer> list = new ArrayList<>();
for (int i = path.size() - 1; i >= 0; i--)
list.add(path.get(i));
return list;
}
}