给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
例如,给出
二叉树:[3,9,20,null,null,15,7]
3
/\
9 20
/ \
15 7返回的层次遍历结果:
[
[3],
[9, 20],
[15, 7],
]
来源:leetcode - 102
知识点:
- 队列:先进先出
- BFS:采用队列保存节点
简要思路:不断向队列中添加节点,进队,出队,直到树中所有的节点都遍历完。
详细思路:定义一个队列,将根节点放入队列,判断队列是否为空,while 循环中添加一个 for 循环,每次添加 n 个元素到 res 结果集中,直到多有节点都遍历完。
代码:
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> res = new ArrayList<>();
Queue<TreeNode> queue = new ArrayList<>();
if (root != null) {
queue.add(root);
}
while (!queue.isEmpty()) {
int n = queue.size();
List<Integer> level = new ArrayList<>();
for (int i = 0; i < n; i++) {
TreeNode node = queue.poll();
level.add(node.val);
if (node.left != null) {
queue.add(node.left);
}
if (node.right != null) {
queue.add(node.right);
}
}
res.add(level);
}
return res;
}
}