AI 助理
备案控制台
开发者社区 开发与运维 文章 正文

[LeetCode算法]有了二叉树层序遍历，妈妈再也不用担心我不会做二叉树层级题了

2023-05-30 159
版权
版权声明：
本文内容由阿里云实名注册用户自发贡献，版权归原作者所有，阿里云开发者社区不拥有其著作权，亦不承担相应法律责任。具体规则请查看《 阿里云开发者社区用户服务协议》和 《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容，填写 侵权投诉表单进行举报，一经查实，本社区将立刻删除涉嫌侵权内容。
简介： 博主最近在刷`leetcode`，做到二叉树套题的时候发现很多题的解题思路都是基于二叉树的层序遍历来完成的，因此写下这篇文章，记录一下二叉树层序遍历这件"神器"在实战的运用。

您好，如果喜欢我的文章，可以关注我的公众号「量子前端」，将不定期关注推送前端好文~

前言

博主最近在刷leetcode，做到二叉树套题的时候发现很多题的解题思路都是基于二叉树的层序遍历来完成的，因此写下这篇文章，记录一下二叉树层序遍历这件"神器"在实战的运用。

[leetcode] 102.二叉树的层序遍历

leetcode题目链接

image.png

二叉树的层序遍历与传统的前序、中序、后序遍历都有一些区别，他是按层级、从左到右、从上到下进行遍历的，因此当我在遍历当前层节点的时候，肯定需要记录当前层所有节点的left、right，保存到队列中，进行下一轮遍历，直到节点没有left、right，则代表已经遍历到了最后一层了。

因此需要借助一个辅助数据结构——队列，队列先进后出，符合层序遍历的顺序性，其实此题就是队列 + 广度优先遍历 的一道结合题。

直接看代码吧：

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number[][]}
 */
var levelOrder = function(root) {
   
   
    const res = [], queue = [];
    queue.push(root);
    if(root === null) return res;

    while(queue.length !== 0) {
   
   
        let level = [];
        const length = queue.length
        for(var i = 0; i < length; i++) {
   
   
            var node = queue.shift();
            level.push(node.val);
            node.left && queue.push(node.left);
            node.right && queue.push(node.right);
        }
        res.push(level);
    }
    return res;
};

接下来我们逐行分析代码。

  • 首先定义了一个结果和一个队列，对应res和queue，将顶层的root节点加入到队列中，开启循环。
  • 在每一轮while 循环中，我们从左到右依次取出节点(shift api)并且判断每个节点下一层是否有后代(left、right)的判断，如果有，则加入到队列中。
  • const length = queue.length记录了队列在每一层遍历开始时的最初状态，保证了后面的for循环遍历的内容是当前层的节点，不会因为left、right加入到队列中的节点影响到当前层的循环轮数。
  • 最终，队列中所有节点都遍历完毕，在for循环中也没有发现新的下层节点，循环结束，返回结果。

此时我们就掌握了二叉树的层序遍历了，那么如下九道力扣上的题目，只需要修改模板的两三行代码（不能再多了），便可打倒！你真的会发现，理解了层序遍历后，解决这些关联题，会如鱼得水一般简单

  • 102.二叉树的层序遍历
  • 107.二叉树的层次遍历II
  • 199.二叉树的右视图
  • 637.二叉树的层平均值
  • 429.N叉树的前序遍历
  • 515.在每个树行中找最大值
  • 116.填充每个节点的下一个右侧节点指针
  • 117.填充每个节点的下一个右侧节点指针II
  • 104.二叉树的最大深度
  • 111.二叉树的最小深度

[leetcode] 107.二叉树的层序遍历II

leetcode题目链接

image.png

此题与102.二叉树的层序遍历极其相似，只需要把最后的res结果的排列顺序改变一下即可，代码架构和102完全一样。

代码:

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number[][]}
 */
var levelOrderBottom = function(root) {
   
   
    var res = [], queue = [];
    queue.push(root);
    if(root === null) return res;
    while(queue.length) {
   
   
        let length = queue.length;
        const level = [];
        for(var i = 0; i < length; i++) {
   
   
            var node = queue.shift();
            level.push(node.val)
            node.left && queue.push(node.left);
            node.right && queue.push(node.right);
        }
        res.unshift(level);
    }
    return res;
};

[leetcode] 199.二叉树的右视图

leetcode题目链接

image.png

此题从题目描述中可以看到，需要收集每一层的最后一个节点，有了"神器"的你，此时已经有思路了吧？这不是只需要在每一轮while循环中的for里加一个判断条件，取出最后一个节点就好了？上代码！

代码:

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number[]}
 */
var rightSideView = function(root) {
   
   
    var res = [], queue = [];
    queue.push(root);
    if(root === null ) return res;
    while(queue.length !== 0 ){
   
   
        const length = queue.length;
        for(var i = 0; i < length; i++) {
   
   
            var node = queue.shift();
            if(i === length - 1) {
   
   
                res.push(node.val);
            }
            node.left && queue.push(node.left);
            node.right && queue.push(node.right);
        }
    }
    return res;
};

[leetcode] 637.二叉树的层平均值

leetcode题目链接

image.png

此题只需要在每层节点取完以后，对节点的平均值进行一个计算即可，相比于前面几题，区别在收集的返回结果不一样，解题代码架构没有区别。

代码:

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number[]}
 */
var averageOfLevels = function(root) {
   
   
    var res = [], queue = [];
    queue.push(root);
    if(root === null) return res;

    while(queue.length !== 0) {
   
   
        const length = queue.length;
        let total = 0;
        for(var i = 0; i < length; i++) {
   
   
            let node = queue.shift();
            total += node.val;
            node.left && queue.push(node.left);
            node.right && queue.push(node.right);
        }
        res.push(total / length);
    }
    return res;
};

[leetcode] 429.N叉树的层序遍历

leetcode题目链接

image.png

此题会有一点小弯需要绕一下，首先数据结构不同，TreeNode节点是这样的：

/**
 * // Definition for a Node.
 * function Node(val,children) {
 *    this.val = val;
 *    this.children = children;
 * };
 */

也就是说我们在每一层节点的循环中，不是再去收集节点的left、right了，而是去遍历节点的children，将children中的节点加入到队列中。

代码:

/**
 * // Definition for a Node.
 * function Node(val,children) {
 *    this.val = val;
 *    this.children = children;
 * };
 */

/**
 * @param {Node|null} root
 * @return {number[][]}
 */
var levelOrder = function(root) {
   
   
    var res = [], queue = [];
    queue.push(root);
    if(root === null) return res;

    while(queue.length !== 0) {
   
   
        var level = [];
        var length = queue.length;
        for(var i = 0; i < length; i++) {
   
   
            var node = queue.shift();
            level.push(node.val)
            for(var item of node.children) {
   
   
                item && queue.push(item);
            }
        }
        res.push(level);
    }
    return res;
};

[leetcode] 515. 在每个树行中找最大值

leetcode题目链接

image.png

此题类似于637.二叉树的层平均值，只是每一层收集的内容变成了最大值。

代码:

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number[]}
 */
var largestValues = function(root) {
   
   
    var res = [], queue = [];
    queue.push(root);
    if(root === null) return res;

    while(queue.length !== 0) {
   
   
        const length = queue.length;
        const list = [];
        for(var i = 0; i < length; i++) {
   
   
            var node = queue.shift();
            list.push(node.val);
            node.left && queue.push(node.left);
            node.right && queue.push(node.right);
        }
        res.push(Math.max(...list));
    }
    return res;
};

[leetcode] 116. 填充每个节点的下一个右侧节点指针

leetcode题目链接

image.png

此题无需重新组装新的返回内容，只需要重新组建root中的每一个节点即可，每一个TreeNode默认的next为null，根据题目要求，在每一层所有节点，我们对除了最右节点以外的所有节点添加一个next属性即可，根据队列先进先出的原则，next的值就是queue[0]，也就是队列的首项。

代码:

/**
 * // Definition for a Node.
 * function Node(val, left, right, next) {
 *    this.val = val === undefined ? null : val;
 *    this.left = left === undefined ? null : left;
 *    this.right = right === undefined ? null : right;
 *    this.next = next === undefined ? null : next;
 * };
 */

/**
 * @param {Node} root
 * @return {Node}
 */
var connect = function(root) {
   
   
    var queue = [root];
    if(root === null) return root;
    while(queue.length) {
   
   
        const length = queue.length;
        for(var i = 0; i < length; i++) {
   
   
            var node = queue.shift();
            if(i < length - 1) {
   
   
                node.next = queue[0];
            }
            node.left && queue.push(node.left);
            node.right && queue.push(node.right);
        }
    }
    return root;
};

[leetcode] 117. 填充每个节点的下一个右侧节点指针II

leetcode题目链接

image.png

此题与116. 填充每个节点的下一个右侧节点指针 类似，直接上代码。

代码:

/**
 * // Definition for a Node.
 * function Node(val, left, right, next) {
 *    this.val = val === undefined ? null : val;
 *    this.left = left === undefined ? null : left;
 *    this.right = right === undefined ? null : right;
 *    this.next = next === undefined ? null : next;
 * };
 */

/**
 * @param {Node} root
 * @return {Node}
 */
var connect = function(root) {
   
   
   if (root === null) {
   
   
        return null;
    }
    let queue = [root];
    while (queue.length > 0) {
   
   
        let n = queue.length;
        for (let i=0; i<n; i++) {
   
   
            let node = queue.shift();
            if (i < n-1) node.next = queue[0];
            if (node.left != null) queue.push(node.left);
            if (node.right != null) queue.push(node.right);
        }
    }
    return root;
};

[leetcode] 104. 二叉树的最大深度

leetcode题目链接

image.png

此题比较简单，只需要在遍历的过程中不断记录height即可，当层序遍历结束，返回height就解决了。

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number}
 */
var maxDepth = function(root) {
   
   
    if(root === null) return root;
    var queue = [root];
    var height = 0;
    while(queue.length > 0) {
   
   
        const length = queue.length;
        height++;
        for(var i = 0; i < length; i++) {
   
   
            var node = queue.shift();
            node.left && queue.push(node.left);
            node.right && queue.push(node.right);
        }
    }
    return height
};

[leetcode] 111. 二叉树的最小深度

leetcode题目链接

image.png

此题与104. 二叉树的最大深度类似，区别在于需要提前结束循环，通过判断树节点是否满足node.left === null && node.right === null，就可以知道二叉树的最小深度是哪个节点，将该节点遍历时的height返回即可。

代码:

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number}
 */
var minDepth = function(root) {
   
   
    if(root === null) return root;
    var queue = [root];
    var height = 0;

    while(queue.length > 0) {
   
   
        const n = queue.length;
        height++;
        for(var i = 0; i < n; i++) {
   
   
            var node = queue.shift();
            if(node.left === null && node.right === null ) {
   
   
                return height;
            }
            node.left && queue.push(node.left);
            node.right && queue.push(node.right);
        }
    }
    return height;
};

结尾

二叉树的层序遍历本质上是队列+广度优先遍历的结合运用诞生出来的"神器"，此时就会发现通过它可以解决leetcode中很多二叉树的题目！

如果本文对你有帮助，不妨点个赞吧~

文章标签：
前端开发
API
算法
关键词：
LeetCode算法
算法leetcode
算法二叉树
LeetCode二叉树层序遍历
LeetCode层序遍历
sorryhc
目录
相关文章
RossyYan
|
11天前
|
存储 算法 测试技术
【C++数据结构——树】二叉树的遍历算法（头歌教学实验平台习题） 【合集】
本任务旨在实现二叉树的遍历，包括先序、中序、后序和层次遍历。首先介绍了二叉树的基本概念与结构定义，并通过C++代码示例展示了如何定义二叉树节点及构建二叉树。接着详细讲解了四种遍历方法的递归实现逻辑，以及层次遍历中队列的应用。最后提供了测试用例和预期输出，确保代码正确性。通过这些内容，帮助读者理解并掌握二叉树遍历的核心思想与实现技巧。
RossyYan
34 2
小山那边
|
25天前
|
存储 算法 Python
文件管理系统中基于 Python 语言的二叉树查找算法探秘
在数字化时代，文件管理系统至关重要。本文探讨了二叉树查找算法在文件管理中的应用，并通过Python代码展示了其实现过程。二叉树是一种非线性数据结构，每个节点最多有两个子节点。通过文件名的字典序构建和查找二叉树，能高效地管理和检索文件。相较于顺序查找，二叉树查找每次比较可排除一半子树，极大提升了查找效率，尤其适用于海量文件管理。Python代码示例包括定义节点类、插入和查找函数，展示了如何快速定位目标文件。二叉树查找算法为文件管理系统的优化提供了有效途径。
小山那边
50 5
光明顶阳顶天
|
2月前
|
算法
分享一些提高二叉树遍历算法效率的代码示例
这只是简单的示例代码，实际应用中可能还需要根据具体需求进行更多的优化和处理。你可以根据自己的需求对代码进行修改和扩展。
光明顶阳顶天
69 8
dufadayang
|
2月前
|
存储 缓存 算法
如何提高二叉树遍历算法的效率？
选择合适的遍历算法，如按层次遍历树时使用广度优先搜索（BFS），中序遍历二叉搜索树以获得有序序列。优化数据结构，如使用线索二叉树减少空指针判断，自定义节点类增加辅助信息。利用递归与非递归的特点，避免栈溢出问题。多线程并行遍历提高速度，注意线程安全。缓存中间结果，避免重复计算。预先计算并存储信息，提高遍历效率。综合运用这些方法，提高二叉树遍历算法的效率。
dufadayang
77 5
小王老师呀
|
2月前
|
存储 算法 Java
leetcode算法题-有效的括号(简单)
【11月更文挑战第5天】本文介绍了 LeetCode 上“有效的括号”这道题的解法。题目要求判断一个只包含括号字符的字符串是否有效。有效字符串需满足左括号必须用相同类型的右括号闭合，并且左括号必须以正确的顺序闭合。解题思路是使用栈数据结构，遍历字符串时将左括号压入栈中，遇到右括号时检查栈顶元素是否匹配。最后根据栈是否为空来判断字符串中的括号是否有效。示例代码包括 Python 和 Java 版本。
小王老师呀
54 4
mao29955
|
2月前
|
机器学习/深度学习 JSON 算法
二叉树遍历算法的应用场景有哪些？
【10月更文挑战第29天】二叉树遍历算法作为一种基础而重要的算法，在许多领域都有着不可或缺的应用，它为解决各种复杂的问题提供了有效的手段和思路。随着计算机科学的不断发展，二叉树遍历算法也在不断地被优化和扩展，以适应新的应用场景和需求。
mao29955
73 0
星霜旅人
|
3月前
|
算法
每日一道算法题（Leetcode 20）
每日一道算法题（Leetcode 20）
星霜旅人
40 2
java冯坚持
|
3月前
|
存储 算法 关系型数据库
数据结构与算法学习二一：多路查找树、二叉树与B树、2-3树、B+树、B*树。（本章为了解基本知识即可，不做代码学习）
这篇文章主要介绍了多路查找树的基本概念，包括二叉树的局限性、多叉树的优化、B树及其变体（如2-3树、B+树、B*树）的特点和应用，旨在帮助读者理解这些数据结构在文件系统和数据库系统中的重要性和效率。
java冯坚持
39 0
数据结构与算法学习二一：多路查找树、二叉树与B树、2-3树、B+树、B*树。（本章为了解基本知识即可，不做代码学习）
java冯坚持
|
3月前
|
存储 算法 搜索推荐
数据结构与算法学习十七：顺序储存二叉树、线索化二叉树
这篇文章主要介绍了顺序存储二叉树和线索化二叉树的概念、特点、实现方式以及应用场景。
java冯坚持
50 0
数据结构与算法学习十七：顺序储存二叉树、线索化二叉树
yantuguiguziPGJ
|
4月前
|
Unix Shell Linux
LeetCode刷题 Shell编程四则 | 194. 转置文件 192. 统计词频 193. 有效电话号码 195. 第十行
本文提供了几个Linux shell脚本编程问题的解决方案，包括转置文件内容、统计词频、验证有效电话号码和提取文件的第十行，每个问题都给出了至少一种实现方法。
yantuguiguziPGJ
71 6
LeetCode刷题 Shell编程四则 | 194. 转置文件 192. 统计词频 193. 有效电话号码 195. 第十行

热门文章

最新文章

  • 1
    基于改进遗传优化的BP神经网络金融序列预测算法matlab仿真
  • 2
    局域网屏幕监控系统中的Python数据结构与算法实现
  • 3
    机器学习算法的优化与改进：提升模型性能的策略与方法
  • 4
    室内障碍物射线追踪算法matlab模拟仿真
  • 5
    排队算法的matlab仿真,带GUI界面
  • 6
    婚恋交友系统平台 相亲交友平台系统 婚恋交友系统APP 婚恋系统源码 婚恋交友平台开发流程 婚恋交友系统架构设计 婚恋交友系统前端/后端开发 婚恋交友系统匹配推荐算法优化
  • 7
    单目标问题的烟花优化算法求解matlab仿真,对比PSO和GA
  • 8
    员工上网行为监控中的Go语言算法：布隆过滤器的应用
  • 9
    强化学习之父Richard Sutton给出一个简单思路，大幅增强所有RL算法
  • 10
    调研180多篇论文，这篇综述终于把大模型做算法设计理清了
  • 1
    155. 最小栈 力扣 python 空间换时间 o(1) 腾讯面试题
    66
  • 2
    1124. 表现良好的最长时间段 (python) 前缀和 分类讨论 最大长度 力扣 面试题
    70
  • 3
    经典的滑动窗口的题目 力扣 2799. 统计完全子数组的数目（面试题）
    65
  • 4
    刷算法Leetcode---9（二叉树篇Ⅲ）
    44
  • 5
    2670.找出不同元素数目差数组-力扣(LeetCode)
    48
  • 6
    821.字符的最短距离-力扣(LeetCode)
    49
  • 7
    力扣经典150题第六十题：反转链表 II
    46
  • 8
    力扣经典150题第五十九题： 随机链表的复制
    53
  • 9
    力扣经典150题第五十八题：合并两个有序链表
    51
  • 10
    力扣经典150题第五十六题：环形链表
    41

    • 相关课程

    更多
  • 智能运维赛（复赛）：利用数据和算法，快速定位系统异常并进行根因分析
  • 智能创作赛（复赛）：相册应用中的视频故事生成算法介绍
  • 智能创作赛（初赛）：相册应用中的故事生成算法介绍
  • 相册服务中的故事生成算法介绍
  • Go语言核心编程 - 数据结构和算法
  • 神经网络概览及算法详解

    • 相关电子书

    更多
  • 数据+算法定义新世界
  • 袋鼠云基于实时计算的反黄牛算法
  • Alink：基于Apache Flink的算法平台

    • 相关实验场景

    更多
  • 使用Swing算法实现商品推荐
  • RSA密码算法设计与实现
  • RSA非对称加密算法
  • 欧拉图的构造性证明与算法实现
  • 推荐系统入门之使用ALS算法实现打分预测
    • 下一篇
    巧用对象存储回源绕过SSRF限制