二叉树的层次遍历使用队列来实现。
如果要一层一层进行输出,那么就需要通过记录上一层的大小来进行输出。
leetcode102题
- 一开始先处理root为空的情况。
- 将头节点如队列,
- 然后一个while循环,结束条件为队列为空。
- 每次循环获取队列的大小size,这个大小就是二叉树某一层的大小。
- 然后循环size次,将队列的全部节点取出,将当前节点输出,并把非空孩子节点加入队列。
- 这样循环size次就得到某一层的全部节点了。
- 一直到队列为空,代表遍历完成。
代码:
import java.util.*;
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> lists=new ArrayList<>();
if(root==null){
return lists;
}
LinkedList<TreeNode> queue=new LinkedList<>();
queue.offer(root);
while(queue.size()!=0){
int size=queue.size();
ArrayList<Integer> list=new ArrayList<>();//存放某一层的节点
for(int i=0;i<size;i++){
TreeNode node=queue.poll();
list.add(node.val);
if(node.left!=null){
queue.offer(node.left);
}
if(node.right!=null){
queue.offer(node.right);
}
}
lists.add(list);//循环结束得到某一层的全部节点
}
return lists;
}
}