二叉树的层序遍历
问题
给你一个二叉树,请你返回其按层序遍历得到的节点值。(即逐层地,从左到右访问所有节点)
示例:
二叉树:[3,9,20,null,null,15,7]
结果:
分析
二叉树的层序遍历就是二叉树的广度优先搜索,简单说就是一层一层的搜索,而每一层是从左到右搜索,遍历完每一层再遍历下一层。
可采用队列来实现,队列是先进先出的,在遍历过程中先将根节点入队,然后在后续遍历过程中实现出队操作,并且将出队元素的左右孩子入队,那么出队的顺序就是层次遍历的顺序。
代码
public class LevelOrder {
public List<List<Integer>> levelOrder(TreeNode root) {
Queue<TreeNode> queue=new LinkedList<>();
List<List<Integer>> result=new ArrayList<>();
//根节点入队
if(root!=null) queue.offer(root);
while(!queue.isEmpty()){
int size=queue.size();
List<Integer> list=new ArrayList<>();
for(int i=0;i<size;i++){
//出队
TreeNode node=queue.poll();
list.add(node.val);
//加入出队元素的左右孩子
if(node.left!=null) queue.offer(node.left);
if(node.right!=null) queue.offer(node.right);
}
result.add(list);
}
return result;
}
}