分行从上往下打印二叉树
从上到下按层打印二叉树,同一层的结点按从左到右的顺序打印,每一层打印到一行。
样例
输入如下图所示二叉树[8, 12, 2, null, null, 6, null, 4, null, null, null]
8
/
12 2
/
6
/
4
输出:[[8], [12, 2], [6], [4]]
在不分行的基础上要在遍历的同时要计算清楚当前层有多少个节点并把他们统一到一个list里
public List<List<Integer>> printFromTopToBottom(TreeNode root) {
List<List<Integer>> res= new ArrayList<>();//二维链表来保存结果
if(root==null){//鲁棒
return res;
}
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
TreeNode head = null;
int nextnodecount = 1;
int nodecount = 0;
while(!queue.isEmpty()){
nodecount = nextnodecount;//把之前的next给现在的
nextnodecount = 0;//从新开始计数
List<Integer> list = new ArrayList<>();//定义这一层的数据集合,根据层数的累加,每次都在更新
for(int i = 1;i<=nodecount;i++){
head = queue.poll();
list.add(head.val);//添加元素到这一层的集合
if(head.left!=null){
queue.offer(head.left);
nextnodecount++;
}
if(head.right!=null){
queue.offer(head.right);
nextnodecount++;
}
}
res.add(list);//把这一层的元素集合给最后的结果集合
}
return res;
}