leetcode 102 二叉树的层序遍历 JavaScript
4个月没刷 层序遍历都差点没敲出来…特此记录
思路: BFS+队列
首先层序遍历要求我们一行一行读取 从左往右将节点进行遍历,这时需要用到队列(先进先出)
每次放进队列queue的都是同一层节点,下一次循环queue就会换成下一层的节点内容
但是queue在遍历某一层的过程中,就要顺便把他的左右子节点push进队列queue中,所以这里需要一些巧妙的方法去解决(在代码中会有备注)
/**
* @param {TreeNode} root
* @return {number[][]}
*/
var levelOrder = function (root) {
if (root == null) return []
const res = [];
const queue = [];
queue.push(root);
// 终止条件
while (queue.length !== 0) {
res.push([]);
// 这里要记录每层进去之前要遍历的节点个数 原因在下一个注释
const thisLevelNodeNum = queue.length;
// 遍历该层所有节点 这里i < queue.length是不行的 因为里面有queue的push操作
for (let i = 0; i < thisLevelNodeNum; i++) {
// 每遍历一个 就卸一个节点
const node = queue.shift();
if (node.left) queue.push(node.left);
if (node.right) queue.push(node.right);
// 最外层的while相当于是层次,每次都会新加一个空数组进res,所以我们要在这个空数组里加入内容 而空数组永远是最后一个
// 所以是length-1
res[res.length-1].push(node.val);
}
}
return res
};