f题目:给你二叉树的根节点发 ,返回其节点值的 层序遍历 。 (即逐层地,从左到右访问所有节点)。root
以下是个人做这道题时的一些理解
解题思路:题目要求层序遍历二叉树,输出结果是可以区分每一层的元素
我们使用广度优先遍历(BFS)输出的结果是[3 9 20 15 7],可见与层序遍历输出结果是一样的,但是题目要求层序遍历可以区分开每一层的遍历结果,如:[[3],[9,20],[15,7]],要如何处理呢?
首先我们要了解广度优先遍历(BFS)是怎么遍历二叉树的:
实现广度优先我们借助了队列这一数据结构,先将要遍历的二叉树的根节点添加到队列,为实现BFS做准备条件,然后在出队列与入队列的循环迭代中,来实现层序遍历
层序遍历的某个过程:
可见在第一层的3节点还未完全出栈,节点2的左右节点(第三层)已经入队列了,这样我们在出队列时就无法使每一层的元素分开,所以,在每一次出队列之前,我们应该获取到此时队列的长度length,每次出队列的次数就是length次,这样就能保证一次性把同一层的所有元素出队列完毕后不在出队列,从而混入下一层的元素, 这样就实现了每层元素分离的效果。
完整代码如下:
public List<List<Integer>> levelOrder(TreeNode root) {
//先定义一个存放整个层序遍历结果的list
List<List<Integer>> res = new ArrayList<>();
if(root == null){
return new ArrayList<>();
}
//定义队列
Queue<TreeNode> queue = new LinkedList<TreeNode>();
//将root节点入队列,为BFS做准备
queue.add(root);
while(!queue.isEmpty()){
//获取每次出队列之前的队列长度
int count = queue.size();
//定义list用来存放每次出队列的TreeNode
List<Integer> list = new ArrayList<Integer>();
//循环控制每次队列的次数,实现将每层元素分离
for(int i = 0; i < count; i++){
//出队列
TreeNode node = queue.poll();
list.add(node.val);
//将poll出的节点的左右节点(下一层入队列),此时丢列先进先出,后进后出的特点实现了层序遍历(也是实现了BFS)
if(node.left != null){
queue.add(node.left);
}
if(node.right != null){
queue.add(node.right);
}
}
//将这次得到的层序遍历结果存入res中
res.add(list);
}
return res;
}