题目描述:
给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
示例:
二叉树:[3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层序遍历结果:
[
[3],
[9,20],
[15,7]
]
解题思路:
可以使用广搜来完成该问题。我们先将根节点添加到队列中,然后将其加入到相应的层次列表中,同时将该节点的孩子节点加到队列中,然后重复循环,直到最终遍历完所有点。
代码:
public class LC102 {
//变型的BFS
//时间复杂度O(n) 空间复杂度O(n)
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> ansList = new ArrayList<>();
if (root == null){
return ansList;
}
//使用队列来存储相应节点
Queue<TreeNode> queue = new LinkedList<>();
//将根节点加入队列
queue.add(root);
while (!queue.isEmpty()){
//获取每一层的元素
List<Integer> level = new ArrayList<>();
//获取当前队列中存在的元素个数
int size = queue.size();
//然后将队列的中的元素分别取出来,放到层次列表中,并将它非空的左右孩子入队
for (int i = 1; i <= size; 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);
}
}
//最后将该层的层次列表存到结果列表中
ansList.add(level);
}
return ansList;
}
}