-
Notifications
You must be signed in to change notification settings - Fork 1
/
Binary_102.java
52 lines (49 loc) · 1.59 KB
/
Binary_102.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
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
public class Binary_102 {
public class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) { val = x; }
}
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> res = new ArrayList<>();
if (root == null)
return res;
LinkedList<TreeNode> q = new LinkedList<>();
q.add(root);
while (!q.isEmpty()) {
List<Integer> level = new ArrayList<>();
int len = q.size();
for (int i=0; i<len; i++) {
TreeNode tmp = q.removeFirst();
level.add(tmp.val);
if (tmp.left != null)
q.add(tmp.left);
if (tmp.right != null)
q.add(tmp.right);
}
res.add(level);
}
return res;
}
// Recursion. Similar with 107.
/*
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> wrapList = new LinkedList<List<Integer>>();
levelMaker(wrapList, root, 0);
return wrapList;
}
public void levelMaker(List<List<Integer>> list, TreeNode root, int level) {
if (root == null) return;
if (level >= list.size()) list.add(new LinkedList<Integer>());
list.get(level).add(root.val);
levelMaker(list, root.left, level+1);
levelMaker(list, root.right, level+1);
// list.get(level).add(root.val);
// Either pre-order or post-order is OK here.
}
*/
}