题目
从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。
思路
主要考察的还是层次遍历二叉树的算法,但是我觉得难点是在需要按层输出上面。
看了解析,觉得在 while 循环里嵌套一个 for 循环非常妙,一开始也有想到for循环,可是却卡在了不知道如何判断层数上。
代码实现
while 循环里嵌套的 for 循环,在一次 while 循环结束后的辅助队列中的 size() 就是当前层数的元素个数,但是如果是用从低到高的 for 循环写法,辅助队列的 size() 会变,并且也不再表示当前层数的元素个数了。所以采用从高到低的 for 循环写法,提前定死边界。
public List<List<Integer>> levelOrder(TreeNode root) {
//创建层次遍历存储List
List<List<Integer>> levelOrderTree = new LinkedList<>();
//创建辅助队列
Queue<TreeNode> bfsTree = new LinkedList<>();
//健壮性检验
if(root != null){
bfsTree.offer(root);
}
while(!bfsTree.isEmpty()){
//创建每一层的存储List
List<Integer> bfsList = new LinkedList<>();
//层次遍历 for循环从高到低
for(int i = bfsTree.size();i > 0;i--){
root = bfsTree.poll(); //获得根节点
bfsList.add(root.val); //添加进当前层的遍历List中
if(root.left != null){ //添加左孩子
bfsTree.offer(root.left);
}
if(root.right != null){ //添加右孩子
bfsTree.offer(root.right);
}
}
levelOrderTree.add(bfsList); //结束一层遍历
}
return levelOrderTree;
}
总结
采用从高到低的 for 循环写法,提前定死边界。