看这道题之前可以复习一下层序遍历
这道题我们需要把每一层返回一个list
需要获取把儿子入队之前的队列长度进行遍历
public List<List<Integer>> levelOrder(TreeNode root) {
if (root == null) return new ArrayList<>();
List<List<Integer>> res = new ArrayList<>();
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
while (!queue.isEmpty()){
List<Integer> list = new ArrayList<>();
int size = queue.size();//添加儿子之前获得该层的结点数
for (int i = 0; i < size; i++) {
TreeNode node = queue.remove();
list.add(node.val);
if (node.left!=null) queue.add(node.left);
if (node.right!=null) queue.add(node.right);
}
res.add(list);
}
return res;
}