二叉树层序遍历
LeetCode编号：102
预备知识：队列，先进先出
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
// 结果返回集合
LinkedList<List<Integer>> res = new LinkedList<>();
// 每层队列，用于临时存每层的节点
Queue<TreeNode> queue = new LinkedList();
if (root == null) {
return res;
}
queue.add(root);
while (!queue.isEmpty()) {
// 每层遍历后的节点集合
LinkedList<Integer> subList = new LinkedList<>();
int nodeSize = queue.size();
// 根据节点数量决定遍历次数，每次遍历一个节点，每层一个list
for (int i = 0; i < nodeSize; i++) {
// 取出节点
TreeNode node = queue.poll();
if (node != null) {
subList.add(node.val);
if (node.left != null) {
queue.offer(node.left);
}
if (node.right != null) {
queue.offer(node.right);
}
}
}
// 每层一次放入结果集合
res.add(subList);
}
return res;
}
}
提交验证
