题目描述:
给你二叉树的根节点 root ,返回其节点值的 层序遍历 。 (即逐层地,从左到右访问所有节点)。
解题思路:
使用队列
先进先出
的特点存储每次遍历的节点。
代码实现:
java
class Solution {
List<List<Integer>> res = new ArrayList<>();
public List<List<Integer>> levelOrder(TreeNode root) {
if (root == null) {
return new ArrayList<>();
}
bfs(root);
return res;
}
// 层遍历
private void bfs(TreeNode root){
if (root==null){
return;
}
// 使用 队列 来遍历并保存,每次记录队列的大小size,
// 每次弹出一个,在队尾加入该节点的左右孩子,同时size--
// 当 size=0,说明这一层就结束了,更新size,更新后的 size大小就是下一层节点的个数
Queue<TreeNode> queue = new LinkedList<TreeNode>();
queue.add(root);// 根节点入队列
while (!queue.isEmpty()) {
List<Integer> path = new ArrayList<>();// 每更新遍历一层的时候都要新建一个
int size = queue.size();// 更新 size
while (size>0){
// 弹出一个
TreeNode tmp = queue.poll();
path.add(tmp.val);// path 记录队列弹出的元素
// 左孩子入队列
if (tmp.left != null) {
queue.add(tmp.left);
}
// 右孩子入队列
if (tmp.right != null) {
queue.add(tmp.right);
}
size--;
}
// 出while循环说明这一层更新完了
res.add(path);// 收获结果,不需要新建一个ArrayList<>(),因此每次的都是最新的
}
}
}