从上到下打印二叉树 II
一、题目
从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。
二、示例
例如:
给定二叉树: [3,9,20,null,null,15,7],
返回其层次遍历结果:
提示:
节点总数 <= 1000
三、思路
I. 按层打印: 题目要求的二叉树的 从上至下 打印(即按层打印),又称为二叉树的 广度优先搜索(BFS)。BFS 通常借助 队列 的先入先出特性来实现。
II. 每层打印到一行: 将本层全部节点打印到一行,并将下一层全部节点加入队列,以此类推,即可分为多行打印。
算法流程:
- 特例处理: 当根节点为空,则返回空列表 [] ;
- 初始化: 打印结果列表 res = [] ,包含根节点的队列 queue = [root] ;
- BFS 循环: 当队列 queue 为空时跳出;
- 新建一个临时列表 tmp ,用于存储当前层打印结果;
- 当前层打印循环: 循环次数为当前层节点数(即队列 queue 长度);
- 出队: 队首元素出队,记为 node;
- 打印: 将 node.val 添加至 tmp 尾部;
- 添加子节点: 若 node 的左(右)子节点不为空,则将左(右)子节点加入队列 queue ;
- 将当前层结果 tmp 添加入 res
四、解法
/**
* 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 res = [];
// 辅助队列 将头结点压入队列
let queue = [root];
// 临时表 用于记录每层数据
let tmp = [];
while(queue.length!=0){
// 清空临时表
tmp = [];
// 记录当前层数中的节点数
let num = queue.length;
// 将当前层节点压入临时表 并将下一层的节点压入辅助队列
while(num--){
// 头结点弹出
let head = queue.shift();
// 将值压入临时表
tmp.push(head.val);
// 当左子树不为空压入辅助队列
if(head.left){
queue.push(head.left);
}
// 当右子树不为空压入队列
if(head.right){
queue.push(head.right)
}
}
// 将当前层临时表压入结果列表
res.push(tmp);
}
// 返回结果
return res;
};