思路:手动构建res结果集合,以及队列;
将当前树节点入队;
将已进入本队列的元素取出,再放入临时集合path中;
按照层数添加叶子节点进入队列;
循环以跳出当前层后将path放入res结果集中。
需要注意的是:
队列的尺寸是随着访问的树的层数而变化的,所以队列尺寸必须在内层循环之外设置为变量,再作为循环退出的条件。
class Solution { public List<List<Integer>> levelOrder(TreeNode root) { List<List<Integer>> res =new ArrayList<>(); Queue<TreeNode> queue =new ArrayDeque<>(); if(root!=null) queue.add(root); //判空操作遗忘通不过测试用例!!! while(!queue.isEmpty()){ List<Integer> path =new ArrayList<>(); int n =queue.size(); for(int i =0 ;i<n;i++){ TreeNode node =queue.poll(); path.add(node.val); if(node.left!=null) queue.add(node.left); if(node.right!=null) queue.add(node.right); } res.add(path); } return res; } }