目录
题目概述(简单难度)
题目链接:
点我进入链接
思路与代码
思路展现
这个题目其实就是二叉树的层序遍历(使用队列)+List集合来完成遍历的操作
同时这道题目还要分层打印出来,这也是这道题目的难点,一开始我也没想出来,后来看到了卡神的操作是真的秀,同学们要是想复习二叉树的层序遍历操作可以来看我的这篇博客:
点我进入对应博客
要想使我们的最后的结果是分层打印出来,就需要使用循环,但是这个循环非常的巧妙:
for(int i = queue.size(); i > 0; i--) {
}
注意我们每次统计队列中的长度的时候一定是当前每层的元素都在队列中的时候,这个个数不多不少正是每层元素的个数,并且注意i必须等于queue.size,不能等于其他的任何数字,并且i必须大于0,不能大于等于0.
如果按照i=0;i<queue.size();i++这样的格式来写的话,我们队列queue的长度是不能固定的,相当于不能保证按层输出了,同学们可以自己在脑海模拟一下就知道了.
代码示例
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
Queue<TreeNode> queue = new LinkedList<>();
List<List<Integer>> res = new ArrayList<>();
if(root != null){
queue.add(root);
}
while(!queue.isEmpty()) {
List<Integer> tmp = new ArrayList<>();
//注意这块的循环必须这样写
for(int i = queue.size(); i > 0; i--) {
TreeNode node = queue.poll();
tmp.add(node.val);
if(node.left != null){
queue.add(node.left);
}
if(node.right != null){
queue.add(node.right);
}
}
res.add(tmp);
}
return res;
}
}