LeetCode 102 二叉树层序遍历：广度优先搜索详解

最新推荐文章于 2024-08-10 16:52:26 发布

Lentr0py

最新推荐文章于 2024-08-10 16:52:26 发布

阅读量259

点赞数 14

文章标签： leetcode 算法 java

本文链接：https://blog.csdn.net/Lentr0py/article/details/141088876

版权

LeetCode 102 二叉树层序遍历：广度优先搜索详解

102. 二叉树的层序遍历

102. 二叉树的层序遍历

题目来源

链接: 102. 二叉树的层序遍历

题目分析

问题描述

在这道题目中，我们需要对给定的二叉树进行层序遍历，并返回每一层节点值的列表。具体题目描述如下：

题目： 给你一个二叉树，请你返回其按层序遍历得到的节点值。（即逐层地，从左到右访问所有节点）。

题目难度

难度：简单

题目标签

标签：树、广度优先搜索、二叉树

题目限制

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

解题思路

为了解决这个问题，我们可以使用广度优先搜索 (BFS) 算法。BFS 使用队列来跟踪当前层的节点，并逐层遍历树的每个节点。每一层的节点值将被存储在一个列表中，最后返回一个包含所有层的节点值的列表。

核心思想：使用队列来保存每一层的节点，每次处理完一层的所有节点后，再处理下一层的节点。

具体步骤：

将根节点加入队列。
当队列不为空时，循环处理队列中的节点。
每次处理队列中的所有节点，并将它们的子节点加入队列。
将处理过的节点的值收集起来，形成每层的结果。

BFS 算法步骤

初始化队列：将根节点加入队列。
遍历每一层：对于队列中的每一层，记录当前层的节点数，并依次将节点从队列中移出，将其值存入当前层的结果列表中。
加入子节点：将当前节点的左右子节点（如果存在）加入队列，等待下一层遍历。
保存当前层结果：将当前层的结果列表加入最终结果列表中。
返回结果：返回包含所有层节点值的结果列表。

代码实现

以下是实现二叉树层序遍历的Java代码：

/**
 * 102. 二叉树的层序遍历
 * @param root 二叉树根节点
 * @return ans 二叉树按层序遍历得到的节点值
 * @apiNote 思路：递归，广度优先遍历，队列
 */
public List<List<Integer>> levelOrder(TreeNode root) {
    // 结果集
        List<List<Integer>> ans = new ArrayList<>();
        if(root == null){
            return ans;
        }
        // 队列
        Deque<TreeNode> queue = new LinkedList<>();
        // 根节点入队
        queue.add(root);
        // 当队列不为空，则将本层的节点值加入结果集，并将下一层的节点入队
        while(!queue.isEmpty()){
            // 当前层节点值
            List<Integer> vals = new ArrayList<>();
            for(int i = queue.size();i>0;i--){
                TreeNode node = queue.removeFirst();
                vals.add(node.val);
                // 左右子树不为空入队
                if(node.left != null){
                    queue.add(node.left);
                }
                if(node.right != null){
                    queue.add(node.right);
                }
            }
            // 将本层的节点值加入结果集
            ans.add(vals);
        }
        return ans;
}