二叉树的层序遍历使用遍历法。使用queue队列作为辅助。先将根节点放入队列。
内层循环把队列元素都取出来放入集合。并且把集合中的每个元素的左右子节点加入队列。等待下一次循环。
每次内循环结束时就完成了上一层节点的集合,把集合加到返回结果集中,继续下一层内循环。
代码如下
public List<List<Integer>> levelOrder(TreeNode root) {
if(root==null)
return new ArrayList<>();
Queue<TreeNode> queue=new LinkedList<>();
List<List<Integer>> res=new ArrayList<>();
queue.add(root);
while(!queue.isEmpty()){
int levelNum=queue.size();
List<Integer> subList=new ArrayList<>();
for(int i=0;i<levelNum;i++){
TreeNode node=queue.poll();
subList.add(node.val);
if(node.left!=null)
queue.add(node.left);
if(node.right!=null)
queue.add(node.right);
}
res.add(subList);
}return res;
}