题目描述
从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3<br /> / \<br /> 9 20<br /> / \<br /> 15 7<br />返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
提示:
节点总数 <= 1000
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-ii-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题解
/**
* 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 = [];
// 初始根节点深度
root.deep = 0;
let queue = [root];
while (queue.length) {
// 取出队列
let node = queue.shift();
// node为null时直接跳过
if (!node) continue;
let deep = node.deep;
// 初始化结果集
if (!res[deep]) res[deep] = [];
// 插入相应节点数据
res[deep].push(node.val);
// 有左节点
if (node.left) {
// 深度 + 1
node.left.deep = deep + 1;
// 推进队列
queue.push(node.left);
}
// 有右节点
if (node.right) {
// 深度 + 1
node.right.deep = deep + 1;
// 推进队列
queue.push(node.right);
}
}
return 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) {
let res = [];
let queue = [root];
while (queue.length) {
let temp = [];
// 对当前整一层的节点进行处理
for (let i = queue.length; i > 0; i--) {
let node = queue.shift();
if (node) {
temp.push(node.val);
queue.push(node.left, node.right);
}
}
if (temp.length > 0) res.push(temp);
}
return res;
};
