【算法】求二叉树的层序遍历

最新推荐文章于 2024-10-26 14:47:35 发布

原创最新推荐文章于 2024-10-26 14:47:35 发布 · 227 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#javascript #算法 #队列 #二叉树

算法和手写代码专栏收录该内容

47 篇文章

订阅专栏

本文介绍了一种实现二叉树层序遍历的方法，通过使用队列来存储每层节点并依次遍历，最终形成一个二维数组，用于展示二叉树各层节点的值。

给定一个二叉树，返回该二叉树层序遍历的结果，（从左到右，一层一层地遍历）
在这里插入图片描述
该二叉树层序遍历的结果是
[
[3],
[9,20],
[15,7]
]

单独从结果上来看，这是个二维数组，我们可以先定义一个res = []作为存储最后结果，
中间还需要定义一个 arr = []用来存储每一层的遍历值

/*
 * function TreeNode(x) {
 *   this.val = x;
 *   this.left = null;
 *   this.right = null;
 * }
 */

/**
  * 
  * @param root TreeNode类 
  * @return int整型二维数组
  */
function levelOrder( root ) {
    if(!root) return [];  //节点为空直接返回空数组
    let res = [];  // 存储最终结果
    let q = [root]  // 定义一个队列  初始值为root 
    // 队列是先进先出的，从左边开始遍历，最先出来的也是最左边的

	// 把每一层的节点都放到一个队列中去，一直把一层的节点都遍历完
    while (q.length) {
        let arr = []  // 用来存储每一层遍历后的结果
        let len = q.length;
        for (let i = 0;i < len; i++) {
            let temp = q.shift() // 从左边开始取出节点
            arr.push(temp.val) // 将该节点值push到arr中
            // 如果该节点有左右子节点，那么依次将这些节点放到队列q中
             if(temp.left) q.push(temp.left)
             if(temp.right) q.push(temp.right)
        }
        res.push(arr) // 收集每一层遍历的值
    }
    return res
}
module.exports = {
    levelOrder : levelOrder
};