题目: 从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一个list里。
题目分析
题目的本质是层序遍历,
题解思路
思路: 初始化一个队列,非空的root节点入队列,根据此时队列的长度,依次出队列;出队列的同时将该节点的非空子节点入队列,循环,直到队列为空。
代码
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;
}
}
与我的方法的不同之处就在于循环控制条件,for(int i = queue.size(); i > 0; i--)
,循环控制的i
初始化为queue.size,每次循环-1
即使循环中往队列里加入了新的节点(即下一层的节点),也不会影响上一层节点的输出。
记录我的思路…道阻且长啊
迭代
思路: 初始化一个队列,按照层数将节点依次入队列,随着该层节点依次出队列并将val值存入List中,将这些节点的左右子节点分别入队列,重复此过程,直到队列为空。
- 若是满二叉树的情况下,二叉树的每层的树节点为
2
n
−
1
2^{n-1}
2n−1 (n为层数),因此可用此数作为循环控制的条件,但是这里的二叉树并不一定是满二叉树,入队列时空节点该如何处理?——对于这个问题,其实类似问题在剑指offer28题中出现过,使用
LinkedList
而非ArrayDeque
可以解决这个问题。就是非要把空节点考虑进去… - 在写完代码运行的时候发现有些测试用例通不过,各种调试之后各种针对bug修补之后,形成了以下代码:
public List<List<Integer>> levelOrder(TreeNode root){
Queue<TreeNode> queue = new LinkedList<TreeNode>();
List<List<Integer>> list_all = new LinkedList<List<Integer>>();
if(root==null) {
return list_all;
}
int nodeNum = 1;
queue.add(root);
while(!queue.isEmpty()) {
List<Integer> list_layer = new LinkedList<Integer>();
int flag=0;
for(int i=0;i<nodeNum;i++) {
TreeNode node = queue.poll();
if(node!=null) {
list_layer.add(node.val);
if(node.left!=null) {
queue.add(node.left);
}else flag++;
if(node.right!=null) {
queue.add(node.right);
}else flag++;
}
}
if(list_layer.size()>0) {
list_all.add(list_layer);
}
nodeNum = nodeNum << 1;
if(flag!=0) nodeNum-=flag;
}
return list_all;
}