【数据结构与算法】力扣 102. 二叉树的层序遍历

题目描述

给你二叉树的根节点 root ,返回其节点值的 层序遍历 。 (即逐层地,从左到右访问所有节点)。

示例 1:

复制代码
输入: root = [3,9,20,null,null,15,7]
输出: [[3],[9,20],[15,7]]

示例 2:

复制代码
输入: root = [1]
输出: [[1]]

示例 3:

复制代码
输入: root = []
输出: []

提示:

  • 树中节点数目在范围 [0, 2000]
  • -1000 <= Node.val <= 1000

分析解答

从输出入口分析我们需要做什么。输出为一个二维数组,所以需要一个 res 作为最终结果数组,需要一个个的 temp 作为每一层的数组。

那么如何控制元素的遍历条件呢?使用队列!

队列中只要有元素,就需要循环遍历。为了将当前层的元素添加到 temp 当前层的数组中。而循环的次数刚好就是当前 queue 的长度(当前层元素个数)。将旧的(当前层元素)推入 temp,将新的(子节点)添加到 queue。

注:需要先保存一下 length 因为 在获取子节点的时候会改变 queue.length

js 复制代码
/**
 * 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) {
    let res = []
    let queue = []
    if (!root) {
        return res
    }
    queue.push(root);
    while (queue.length) {
        // 当前层的节点
        let temp = []
        // 需要先保存一下 length 因为 在获取子节点的时候会改变 queue.length
        let length = queue.length
        for (let i = 0; i < length; i++) {
            let node = queue.shift()
            temp.push(node.val)
            node.left && queue.push(node.left)
            node.right && queue.push(node.right)
        }
        res.push(temp)
    }
    return res
};

上图来自:代码随想录

思路拓展

相关推荐
BB_CC_DD17 分钟前
四. 以Annoy算法建树的方式聚类清洗图像数据集,一次建树,无限次聚类搜索,提升聚类搜索效率。(附完整代码)
深度学习·算法·聚类
梁下轻语的秋缘2 小时前
每日c/c++题 备战蓝桥杯 ([洛谷 P1226] 快速幂求模题解)
c++·算法·蓝桥杯
CODE_RabbitV2 小时前
【深度强化学习 DRL 快速实践】逆向强化学习算法 (IRL)
算法
mit6.8242 小时前
[贪心_7] 最优除法 | 跳跃游戏 II | 加油站
数据结构·算法·leetcode
keep intensify2 小时前
通讯录完善版本(详细讲解+源码)
c语言·开发语言·数据结构·算法
shix .2 小时前
2025年PTA天梯赛正式赛 | 算法竞赛,题目详解
数据结构·算法
风铃儿~3 小时前
Java面试高频问题(26-28)
java·算法·面试
wuqingshun3141593 小时前
蓝桥杯 4. 卡片换位
算法·职场和发展·蓝桥杯
江沉晚呤时3 小时前
深入了解C# List集合及两种常见排序算法:插入排序与堆排序
windows·sql·算法·oracle·c#·排序算法·mybatis
Eric.Lee20213 小时前
数据集-目标检测系列- F35 战斗机 检测数据集 F35 plane >> DataBall
人工智能·算法·yolo·目标检测·计算机视觉