给你二叉树的根节点 root ,返回其节点值的 层序遍历 。 (即逐层地,从左到右访问所有节点)。
示例:
输入:root = [3,9,20,null,null,15,7]
输出:[[3],[9,20],[15,7]]
广度优先遍历是按层层推进的方式,遍历每一层的节点。题目要求的是返回每一层的节点值,所以这题用广度优先来做非常合适。
广度优先需要用队列作为辅助结构,我们先将根节点放到队列中,然后不断遍历队列。
首先拿出根节点,如果左子树/右子树不为空,就将他们放入队列中。根节点放入之后,根节点再从队列中拿走了,而根节点的两个孩子放入队列中
我们把每层遍历到的节点都放入到一个结果集中,最后返回这个结果集就可以了。
时间复杂度: O(n)
空间复杂度:O(n)
public List<List<Integer>> levelOrder(TreeNode root) {
if(root == null){
return new ArrayList<List<Integer>>();
}
List<List<Integer>> res = new ArrayList<List<Integer>>();
LinkedList<TreeNode> queue = new LinkedList<>();
//先将根节点放入队列中
queue.add(root);
while (queue.size() > 0){
//获取当前队列的长度,即当前层的节点数
int size = queue.size();
ArrayList<Integer> tmp = new ArrayList<>();
for (int i = 0; i < size; i++) {
//将队列中的元素拿出来(这一层的节点),移除的元素加入到临时集合tmp中
TreeNode remove = queue.remove();
tmp.add(remove.val);
//如果节点的左右子树不空,放入队列中,以便下次循环加入到临时集合中
if (remove.left != null){
queue.add(remove.left);
}
if (remove.right != null){
queue.add(remove.right);
}
}
//将临时集合tmp加入到结果集中
res.add(tmp);
}
return res;
}