二叉树的层序遍历(JS实现)
二叉树的层序遍历(JS实现)
1 题目
给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
示例:
二叉树:[3,9,20,null,null,15,7],
3
/
9 20
/
15 7
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
2 思路
这道题的主要思路是用队列来进行层序遍历,由于需要区分每一层,因此额外使用一个数组来存储下一层的节点
3代码
/**
* 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) {
const result = [];
if (!root) return result;
let quque = [root];
let temp = [], res = []; // temp用于存储下一层级的节点,而res用户存储当前层级的值
let p;
while(quque.length > 0 || temp.length > 0) {
if (quque.length === 0) { // 当队列长度为0,说明当前层级所有节点已经访问过
result.push(res); //保存当前层级的值
quque = temp; //访问下一层
temp = [];
res = [];
}
p = quque.shift();
if (p) {
res.push(p.val);
temp.push(p.left);
temp.push(p.right);
}
}
return result;
};