二叉树的层序遍历,顾名思义是按照二叉树的每一层进行遍历的,如下图的二叉树,它的层序遍历顺序就是:A B C D E F G H
我们发现层序遍历就是按照一层一层的顺序访问的,那么此时就可以联想到队列先进先出的特性,即:对于二叉树从上往下从左往右依次放到队列中,并且遍历一遍即可
。
步骤:
空树
:直接返回二叉树不为空
:将根节点入队列,并循环执行以下操作(循环条件:队列不为空
):- 取队头元素,遍历该元素并且从队列中删除:cur=queue.poll();
- 如果cur有左孩子:入队列
- 如果cur有右孩子:入队列
注意:
在leetcode中的原题返回的是List<List> 类型,也就是把每一层的元素需要装在一个list中,我们在遍历队列中的元素时先获取队列中元素的个数即可,通过个数来控制内循环的次数(也就是每层元素的个数)
代码实现:
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> rest=new ArrayList<>();
if(root==null) return rest;
Queue<TreeNode> queue=new LinkedList<>();
queue.offer(root);//根入队列
while(!queue.isEmpty())
{
List<Integer> tmp=new ArrayList<>();
//得到此时队列中有多少元素,也就循环遍历多少次
int size=q.size();
//访问队列里的元素并将其加入到每一层
while(size>0)
{
//让当前元素出队列
TreeNode cur=queue.poll();
//添加到list中(遍历)
tmp.add(cur.val);
//判断是否有左右孩子
if(cur.left!=null)
queue.offer(cur.left);
if(cur.right!=null)
queue.offer(cur.right);
size--;
}
rest.add(tmp);
}
return rest;
}
}