
求解代码
构建一个辅助队列queue,让root首先进入队列。
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
|
public ArrayList<ArrayList<Integer>> levelOrder (TreeNode root) {
ArrayList<ArrayList<Integer>> ans = new ArrayList<>();
if(root==null){
return ans;
}
Queue<TreeNode> queue = new ArrayDeque<TreeNode>();
queue.add(root);
while(!queue.isEmpty()){
ArrayList<Integer> list = new ArrayList<>();
int n = queue.size();
for(int i=0;i<n;i++){
TreeNode cur = queue.poll();
list.add(cur.val);
if (cur.left!=null) {
queue.add(cur.left);
}
if(cur.right!=null){
queue.add(cur.right);
}
}
ans.add(list);
}
return ans;
}
|