⚡️剑指 Offer 32 - II. 从上到下打印二叉树 II⚡️
从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/
9 20
/
15 7
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
提示:
节点总数 <= 1000
⚡️分析⚡️
数组嵌套数组,内层数组为每一层的总结点数
思路:也是用队列来解决,先判断此时队列长度(每一层的结点个数),然后通过for循环(长度为队列长度),弹出队头,压入临时数组,左或右有子结点那就压入队列,for循环结束后可以保证该层全部结点已经压入临时数组,再把该临时数组压入返回的数组即可。
可以发现“for循环为队列长度”这个设计思想很精妙
代码如下:
/**
* Definition for a binary tree node.
* function TreeNode(val) {
* this.val = val;
* this.left = this.right = null;
* }
*/
/**
* @param {TreeNode} root
* @return {number[][]}
*/
var levelOrder = function(root) {
if(!root) return [];
let arr = [], queue = [], temp = [], node, len;
queue.push(root);
while(queue.length) {
temp = [];
len = queue.length;
for(let i=0; i<len; i++) {
node = queue.shift();
temp.push(node.val);
if(node.left) {
queue.push(node.left);
}
if(node.right) {
queue.push(node.right);
}
}
arr.push(temp);
}
return arr;
};
算法效率如图: