剑指 Offer 32 - II. 从上到下打印二叉树 II
题目描述
从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。
例如:
给定二叉树: [3,9,20,null,null,15,7],
返回其层次遍历结果:
[ [3], [9,20], [15,7] ]
提示
节点总数 <= 1000
思路
BFS
queue数组: 模拟队列
res数组:存放结果
temp数组:存放每一层的节点数据
- root的根结点入队
- 遍历队列 将队列的元素逐一出队 push进temp数组
- 判断是否存在左右子节点 若存在 则将其push 进队列
注意:
queue队列长度会发生改变,在做遍历操作时需要更新遍历终点
代码
/**
* 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 queue = [root];
let res = [];
while(queue.length){
let temp = [];
//更新遍历终点值
let len = queue.length;
//遍历队列
for(let i = 0; i <len ; i++){
//出队
let n = queue.shift();
//将出队元素 push至本层的临时数组 temp
temp.push(n.val);
//若存在左右节点 将左右节点入队
if(n.left) queue.push(n.left);
if(n.right) queue.push(n.right);
}
//遍历一层结束 push进结果数组
res.push(temp);
}
return res;
};