102.二叉树的层序遍历
102.二叉树的层序遍历
题目
给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
示例:
二叉树:[3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层序遍历结果:
[
[3],
[9,20],
[15,7]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-level-order-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题解
层序遍历,可以借助队列
开始队列为空的时候,根节点进去,第一层节点都入队了。
此时根节点出队,根节点的孩子节点入队。
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> res = new ArrayList<>();
Queue<TreeNode> queue = new LinkedList<>();
if(root == null) return res;
queue.offer(root);
while(!queue.isEmpty()){
List<Integer> tmp = new ArrayList<>();
int count = queue.size();//表示每一层有多少个节点,
while(count-->0){ //控制输出一层的节点,并该层节点的孩子节点入队
TreeNode cur = queue.poll();
tmp.add(cur.val);
if(cur.left!=null)queue.offer(cur.left);
if(cur.right!=null)queue.offer(cur.right);
}
res.add(tmp);
}
return res;
}
}