leetcode刷题(剑指offer) 102.二叉树的层序遍历

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

本题是要层级遍历二叉树,将每层的所有节点都存入到list中。主要存在两个需要解决的点。

  • 如何层级遍历二叉树
  • 如何知道每层有哪些节点

接替围绕着这两个点,首先可以使用队列来层级遍历每个节点,思路类似于宽度优先搜索(bfs),先将root节点放入队列中,每次都弹出队头节点,取出节点,然后将取出节点的left节点和right节点依次放入到队列中,直到没有节点再能放入到队列中为止(即:队列为空)。

随后再此基础上,创建变量lastNode表示为当前层的最后一个节点,nextLastNode表示为下一层的最后一个节点,这个变量每次从队列中取出节点都会进行更新,当弹出节点等于lastNode时,即可将一层的list存入总的list,且将lastNode更新为nextLastNode

代码实现如下:

java 复制代码
public static List<List<Integer>> levelOrder(TreeNode root) {
    if (root == null) {
        return new ArrayList<>();
    }
    List<List<Integer>> res = new ArrayList<>();
    TreeNode lastNode = root;
    TreeNode nextLastNode = null;
    Queue<TreeNode> queue = new ArrayDeque<>();
    List<Integer> list = new ArrayList<>();
    queue.add(root);
    while (!queue.isEmpty()) {
        TreeNode node = queue.poll();
        list.add(node.val);
        if (node.left != null) {
            nextLastNode = node.left;
            queue.add(node.left);
        }
        if (node.right != null) {
            nextLastNode = node.right;
            queue.add(node.right);
        }
        if (lastNode == node) {
            res.add(list);
            list = new ArrayList<>();
            lastNode = nextLastNode;
        }

    }
    return res;
}
相关推荐
聚客AI1 天前
🙋‍♀️Transformer训练与推理全流程:从输入处理到输出生成
人工智能·算法·llm
大怪v1 天前
前端:人工智能?我也会啊!来个花活,😎😎😎“自动驾驶”整起!
前端·javascript·算法
CYRUS_STUDIO1 天前
用 Frida 控制 Android 线程:kill 命令、挂起与恢复全解析
android·linux·逆向
熊猫李1 天前
rootfs-根文件系统详解
linux
惯导马工1 天前
【论文导读】ORB-SLAM3:An Accurate Open-Source Library for Visual, Visual-Inertial and
深度学习·算法
骑自行车的码农1 天前
【React用到的一些算法】游标和栈
算法·react.js
博笙困了1 天前
AcWing学习——双指针算法
c++·算法
dessler1 天前
Hadoop HDFS-高可用集群部署
linux·运维·hdfs
moonlifesudo1 天前
322:零钱兑换(三种方法)
算法
泽泽爱旅行1 天前
awk 语法解析-前端学习
linux·前端