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;
}
相关推荐
前端拿破轮2 分钟前
🤡🤡🤡面试官:就你这还每天刷leetcode?连四数相加和四数之和都分不清!
算法·leetcode·面试
cat_with_cat9 分钟前
Linux网络:UDP socket创建流程与简单通信
linux·网络·udp
地平线开发者26 分钟前
征程 6|工具链量化简介与代码实操
算法·自动驾驶
DoraBigHead32 分钟前
🧠 小哆啦解题记——谁偷改了狗狗的台词?
算法
Kaltistss34 分钟前
240.搜索二维矩阵Ⅱ
线性代数·算法·矩阵
2401_8616152838 分钟前
跨平台的ARM 和 x86 Docker 镜像:汇编语言实验环境搭建
linux·汇编·ubuntu·docker·容器
轻语呢喃38 分钟前
每日LeetCode:合并两个有序数组
javascript·算法
hz3557242 分钟前
IO多路复用--epoll
linux
努力学习的小廉43 分钟前
深入了解linux系统—— System V之消息队列和信号量
android·linux·开发语言
华无丽言44 分钟前
如何简单实现发版不影响客户使用?nginx负载
linux·nginx