链接
题目描述
BFS
本题主要是输出结果的格式问题
对于每一行,用一个List
封装,然后把每一个List
结合到一个最后的List
中输出。
可以使用一个 size
变量,记录队列的长度(即为本行的Node
数量)
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> res = new ArrayList<>();
Queue<TreeNode> queue = new LinkedList<>();
if(root == null){
return res;
}
queue.offer(root);
TreeNode cur = null;
while(!queue.isEmpty()){
int size = queue.size();
List<Integer> temp = new ArrayList<>();
while(size > 0){
cur = queue.poll();
temp.add(cur.val);
if(cur.left != null){
queue.offer(cur.left);
}
if(cur.right != null){
queue.offer(cur.right);
}
size--;
}
res.add(temp);
}
return res;
}
}