⚡️剑指 Offer 32 - III. 从上到下打印二叉树 III⚡️
请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其他行以此类推。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[20,9],
[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, flag = 1;
queue.push(root);
while(queue.length) {
temp = [];
len = queue.length;
for(let i=0; i<len; i++) {
node = queue.shift();
if(flag == 1)
temp.push(node.val);
else
temp.unshift(node.val);
if(node.left) {
queue.push(node.left);
}
if(node.right) {
queue.push(node.right);
}
}
arr.push(temp);
flag = !flag;
}
return arr;
};
算法效率如图: