宽度优先搜索算法(BFS)

宽度优先搜索算法(BFS)是什么?

宽度优先搜索算法(BFS) (也称为广度优先搜索 )主要运用于树、图和矩阵(这三种可以都归类在图中),用于在图中从起始顶点开始逐层地向外探索,直到找到目标顶点为止。

在本篇文章中,我们主要讨论其在中的运用

树的宽度优先搜索

树的宽度优先搜索 即 树的层序遍历:逐层访问树的节点,并按照层级顺序输出节点的值。从树的根节点开始,逐层向下遍历,先访问当前层的所有节点,然后再访问下一层的节点,依次类推直到遍历完整棵树

其过程如下图所示:

如何实现树的层序遍历呢?

我们在遍历每一层节点时,都是从左向右 依次遍历的,即先遍历上一层节点中的第一个节点的孩子节点 , 即优先遍历前面的节点,此时,满足**"先进先出"** ,因此,我们可以使用队列来帮助我们完成层序遍历

具体步骤如下:

  1. 先将根节点放入队列中

  2. 从队列中出队一个节点,并将该节点的所有孩子节点(如果有)依次放入队列中

  3. 继续取出节点,指导队列为空

  4. 当队列为空时,遍历完成

接下来,我们通过一些练习来进一步体会和掌握BFS

相关练习

练习1:N叉树的层序遍历

题目链接:

429. N 叉树的层序遍历 - 力扣(LeetCode)

题目描述:

给定一个 N 叉树,返回其节点值的层序遍历。(即从左到右,逐层遍历)。

树的序列化输入是用层序遍历,每组子节点都由 null 值分隔(参见示例)。

示例 1:

复制代码
输入:root = [1,null,3,2,4,null,5,6]
输出:[[1],[3,2,4],[5,6]]

示例 2:

复制代码
输入:root = [1,null,2,3,4,5,null,null,6,7,null,8,null,9,10,null,null,11,null,12,null,13,null,null,14]
输出:[[1],[2,3,4,5],[6,7,8,9,10],[11,12,13],[14]]

提示:

  • 树的高度不会超过 1000
  • 树的节点总数在 [0, 10^4] 之间

思路分析:

本题要求我们返回N叉树节点值的层序遍历,且以 List<List<Integer>>类型的结果返回,即每一层使用一个列表来存储本层的所有节点的值,然后将这些列表放到一个列表中返回

在这里,我们可以使用队列来帮助我们进行层序遍历,但在将节点出队时,也会将其孩子节点放入队列中,那么,我们如何区分每一层的节点呢?

我们可以使用一个变量 count 在节点出队之前,记录队列当前节点个数(即这一层节点个数),然后再将这 count 个节点依次出队列,将其节点值放入列表中,同时将其孩子节点放入队列中

具体步骤:

  1. 先判断树是否为空,不为空将根节点放入队列中

  2. 记录当前队列中包含的节点个数count

  3. 将count个节点依次出队列,将其节点值放入一个临时列表中,同时将其孩子节点放入队列中

  4. 遍历完count个节点后,临时列表中就存放了这一层所有节点的值,然后再将临时列表放入结果列表中

  5. 继续遍历,直到队列为空

  6. 返回结果

代码实现:

java 复制代码
class Solution {
    public List<List<Integer>> levelOrder(Node root) {
       List<List<Integer>> ret = new ArrayList<>();
        Node cur = root;
        Queue<Node> queue = new ArrayDeque<>();
        //先将根节点放入队列中
        if(cur == null) return ret;
        queue.add(cur);

        int count = 0;
        while (!queue.isEmpty()){
            //先记录当前队列中节点个数
            count = queue.size();
            List<Integer> tmp = new ArrayList<>();//使用临时表存储这一层节点的值
            for (int i = 0; i < count; i++) {
                cur = queue.poll();//将队头元素出队
                tmp.add(cur.val);
                for (Node child: cur.children) {//将队头元素的孩子节点放入队列中
                    queue.add(child);
                }
            }
            ret.add(tmp);//将临时表放入ret中
        }
        return ret;
    }
}

练习2:二叉树的锯齿形层序遍历

题目链接:

103. 二叉树的锯齿形层序遍历 - 力扣(LeetCode)

题目描述:

给你二叉树的根节点 root ,返回其节点值的 锯齿形层序遍历 。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。

示例 1:

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

示例 2:

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

示例 3:

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

提示:

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

思路分析:

本题要求我们返回二叉树的锯齿形层序遍历,

什么是锯齿形层序遍历?

先从左向右,再从右向左地对每一层节点进行遍历

以示例1为例:

当前层数为奇数时,从左向右遍历;当前层数为偶数时,从右向左遍历

因此,本题的解题思路与练习1类似,唯一不同的是:在偶数层时需要将所有节点的值进行逆序,我们可以使用一个int类型的变量(或boolean类型的变量)来标记这一层的节点是否需要逆序

代码实现:

java 复制代码
class Solution {
    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
       List<List<Integer>> ret = new ArrayList<>();
        Queue<TreeNode> queue = new ArrayDeque<>();
        int flag = 1;//标识是否需要逆序
        TreeNode cur = root;
        //先将根节点放入队列中
        if(cur == null) return ret;
        queue.add(cur);

        int count = 0;
        while(!queue.isEmpty()){
            //先记录当前队列节点个数
            count = queue.size();
            //依次出队列
            List<Integer> tmp = new ArrayList<>();
            for(int i = 0; i < count; i++){
                cur = queue.poll();
                tmp.add(cur.val);
                //将其孩子节点入队列
                if(cur.left != null) queue.add(cur.left);
                if(cur.right != null) queue.add(cur.right);
            }
            //判断是否需要逆序
            if(flag == -1){
                Collections.reverse(tmp);
            }
            ret.add(tmp);
            flag = -flag;
            
        }
        return ret;
    }
}

练习3:在每个树行中找最大值

题目链接:

LCR 044. 在每个树行中找最大值 - 力扣(LeetCode)

题目描述:

给定一棵二叉树的根节点 root ,请找出该二叉树中每一层的最大值。

示例1:

复制代码
输入: root = [1,3,2,5,3,null,9]
输出: [1,3,9]
解释:
          1
         / \
        3   2
       / \   \  
      5   3   9 

示例2:

复制代码
输入: root = [1,2,3]
输出: [1,3]
解释:
          1
         / \
        2   3

示例3:

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

示例4:

复制代码
输入: root = [1,null,2]
输出: [1,2]
解释:      
           1 
            \
             2     

示例5:

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

提示:

  • 二叉树的节点个数的范围是 [0,104]
  • -231 <= Node.val <= 231 - 1

思路分析:

题目要求我们找到二叉树中每一层中的最大值,因此,我们只需要进行层序遍历,并在每一层中寻找最大值即可

代码实现:

java 复制代码
class Solution {
    public List<Integer> largestValues(TreeNode root) {
        List<Integer> ret = new ArrayList<>();
        TreeNode cur = root;
        Queue<TreeNode> queue = new ArrayDeque<>();
        //先将根节点放入
        if(cur == null) return ret;
        queue.add(cur);

        int count = 0;
        while(!queue.isEmpty()){
            //先计算队列中有多少个元素
            count = queue.size();
            //将元素依次出队列
            int max = Integer.MIN_VALUE;
            for(int i = 0; i < count; i++){
                cur = queue.poll();
                if(max < cur.val){//寻找这一层的最大值
                    max = cur.val;
                }
                //将其孩子节点放入队列中
                if(cur.left != null) queue.add(cur.left);
                if(cur.right != null) queue.add(cur.right);
            }
            ret.add(max);//将这一层的最大值添加到ret中
        }
        return ret;
    }
}

练习4:二叉树的最大宽度

题目链接:

662. 二叉树最大宽度 - 力扣(LeetCode)

题目描述:

给你一棵二叉树的根节点 root ,返回树的 最大宽度

树的 最大宽度 是所有层中最大的 宽度

每一层的 宽度 被定义为该层最左和最右的非空节点(即,两个端点)之间的长度。将这个二叉树视作与满二叉树结构相同,两端点间会出现一些延伸到这一层的 null 节点,这些 null 节点也计入长度。

题目数据保证答案将会在 32 位 带符号整数范围内。

示例 1:

复制代码
输入:root = [1,3,2,5,3,null,9]
输出:4
解释:最大宽度出现在树的第 3 层,宽度为 4 (5,3,null,9) 。

示例 2:

复制代码
输入:root = [1,3,2,5,null,null,9,6,null,7]
输出:7
解释:最大宽度出现在树的第 4 层,宽度为 7 (6,null,null,null,null,null,7) 。

示例 3:

复制代码
输入:root = [1,3,2,5]
输出:2
解释:最大宽度出现在树的第 2 层,宽度为 2 (3,2) 。

提示:

  • 树中节点的数目范围是 [1, 3000]
  • -100 <= Node.val <= 100

思路分析:

题目要求我们返回二叉树的最大宽度,需要注意的是:这里的宽度不是指这一层有多少个节点,而是指这一层最左端的非空节点left和最右端的非空节点right之间的长度,left和right之间为空或不存在的节点也需计入长度

我们以示例2为例:

left和right之间的节点及其延伸到这一层的null节点都要计入,因此,我们首先可以想到:可以添加这些空节点,即若当前节点的右孩子为空,则将null入队列;若当前节点为null,则将null null两个节点入队列,这样,我们就补齐了这一层两端点之间的节点,因此可以直接使用 count 变量记录这一层的宽度,但是,虽然树中节点的数目范围为 1 - 3000,但若出现极端情况,即:

此时,需要补充的空节点数量非常多,超出时间限制,因此,该方法不能解决本题

本层宽度 = 最左端的非空节点left和最右端的非空节点right之间的长度,因此,我们可以直接寻找最左端非空节点left的位置 和最右端非空节点right的位置,然后就可以直接求出这一层的宽度

如何找到最左端非空节点left的位置 和最右端非空节点right的位置?

我们对节点进行编号,从1开始(即根节点为1),这样当前节点cur的编号为index,则其左孩子为2*index,右孩子为2*index + 1,通过编号,我们就能很容易的找到每一个节点,此时, 每一层的宽度 = 最右侧非空节点right的编号 - 最左侧非空节点left的编号 + 1,在这里,我们使用 Pair(配对) 来关联当前节点和其编号

具体步骤:

  1. 将根节点及其编号1放入队列中,此时最大宽度为1

  2. 先记录当前队列中元素个数count

  3. 将count个元素依次出队列,记录第一个节点的编号 和最后一个节点的编号,遍历过程中,将当前节点的孩子节点和编号放入队列中

  4. 计算该层宽度 right - left + 1,并更新最大宽度

  5. 继续遍历,直达队列为空

代码实现:

java 复制代码
class Solution {
    public int widthOfBinaryTree(TreeNode root) {
        TreeNode cur = root;
        Queue<Pair<TreeNode, Integer>> queue = new ArrayDeque<>();
        queue.add(new Pair<TreeNode, Integer>(root, 1));//将根节点添加到队列中
        int ret = 1;//最大宽度
        int count = 0;
        while(!queue.isEmpty()){
            count = queue.size();
            int left = 0, right = 0;//记录最左端和最右端非空节点的编号
            for(int i = 0; i < count; i++){
                Pair<TreeNode, Integer> pair = queue.poll();
                int index = pair.getValue();
                if(i == 0) left = index;//第一个非空节点
                if(i == count - 1) right = index;//最后一个非空节点
                cur = pair.getKey();
                if(cur.left != null) queue.add(new Pair<TreeNode, Integer>(cur.left, index*2));
                if(cur.right != null) queue.add(new Pair<TreeNode, Integer>(cur.right, index*2 + 1));
            }
            ret = Math.max(ret, right - left + 1);//更新最大宽度
        }
        return ret;
    }
}
相关推荐
游是水里的游几秒前
【算法day19】回溯:分割与子集问题
算法
不想当程序猿_1 分钟前
【蓝桥杯每日一题】分糖果——DFS
c++·算法·蓝桥杯·深度优先
晚夜微雨问海棠呀3 分钟前
长沙景区数据分析项目实现
开发语言·python·信息可视化
graceyun3 分钟前
C语言初阶习题【9】数9的个数
c语言·开发语言
hanbarger5 分钟前
mybatis框架——缓存,分页
java·spring·mybatis
cdut_suye13 分钟前
Linux工具使用指南:从apt管理、gcc编译到makefile构建与gdb调试
java·linux·运维·服务器·c++·人工智能·python
南城花随雪。20 分钟前
单片机:实现FFT快速傅里叶变换算法(附带源码)
单片机·嵌入式硬件·算法
苹果醋325 分钟前
2020重新出发,MySql基础,MySql表数据操作
java·运维·spring boot·mysql·nginx
小蜗牛慢慢爬行26 分钟前
如何在 Spring Boot 微服务中设置和管理多个数据库
java·数据库·spring boot·后端·微服务·架构·hibernate
azhou的代码园29 分钟前
基于JAVA+SpringBoot+Vue的制造装备物联及生产管理ERP系统
java·spring boot·制造