题目描述
从上到下按层打印二叉树，同一层结点从左至右输出。每一层输出一行。
题目分析
从上到下打印二叉树我们知道用队列可以实现，但是如果多行打印怎么做呢？
我们需要分割,在行与行之间进行分割。如何分割呢？肯定要知道个数才能分割。
可是我又如何知道这一行有多少个呢？
这就是重点了，我们可以通过遍历上一层，通过它们的子树就可以知道这一层有多少个节点了。
当然我们还需要有一个变量来记录这一层已经打印了多少个节点了。
所以我们需要一个队列+两个个变量。
代码
function Print(pRoot) {
const queue = [],
res = [];
if (pRoot === null) {
return res;
}
queue.push(pRoot);
let nextLevel = 0; // 下一层节点个数
let toBePrinted = 1; // 这一层还有多少个节点要打印
let list = []; // 存放每一层节点
while (queue.length) {
const pNode = queue.shift();
list.push(pNode.val);
if (pNode.left !== null) {
queue.push(pNode.left);
nextLevel++;
}
if (pNode.right !== null) {
queue.push(pNode.right);
nextLevel++;
}
toBePrinted--;
if (toBePrinted === 0) {
res.push(list);
list = [];
toBePrinted = nextLevel;
nextLevel = 0;
}
}
return res;
}