前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >【算法篇】三道题理解算法思想——认识BFS

【算法篇】三道题理解算法思想——认识BFS

作者头像
小皮侠
发布2024-04-08 20:49:37
1060
发布2024-04-08 20:49:37
举报

BFS(宽搜)

宽度优先遍历和深度优先遍历组成了大家熟悉的搜索算法,这两种算法也是蓝桥杯之类竞赛题的常考思想,正巧马上蓝桥杯临近,博主也是刷了很多BFS相关的题型,在这篇文章中会从力扣上选取三道简单的宽搜题型,带大家了解BFS的模板以及对他有个初步认识。 本篇文章题目较为简单,大家可以根据第一题的模板,自己先去力扣上做题然后回来看题解,稍后我们继续更新难度更高的宽搜题目,希望大家能给个关注👍。

文章顺序:

题目链接-》算法思路-》代码呈现。

算法摘要:

宽度优先遍历是一种利用队列这种数据结构,从某一点开始,一层一层进行遍历的一种算法思想,而BFS(宽搜)实际上就是一种暴力搜索算法,利用宽度优先遍历来查找想要结果。

1.N叉树的层序遍历

题目链接:

https://leetcode.cn/problems/n-ary-tree-level-order-traversal/description/

算法思路:

仅需多加⼀个变量,⽤来记录每⼀层结点的个数,然后层序遍历即可。

代码展示:

代码语言:javascript
复制
/*
// Definition for a Node.
class Node {
    public int val;
    public List<Node> children;

    public Node() {}

    public Node(int _val) {
        val = _val;
    }

    public Node(int _val, List<Node> _children) {
        val = _val;
        children = _children;
    }
};
*/

class Solution {
    public List<List<Integer>> levelOrder(Node root) {
        List<List<Integer>> lists=new ArrayList<>();
        if(root==null){
            return lists;
        }
        Queue<Node> q=new LinkedList<Node>();
        q.add(root);
        while(!q.isEmpty()){
            int sz=q.size();
            List<Integer> list=new ArrayList<>();
            for(int i=0;i<sz;i++){
                Node cur=q.poll();
                list.add(cur.val);
                for(Node c:cur.children){
                    if(c!=null){
                        q.add(c);
                    }
                }
            }
            lists.add(list);
        }
      return lists;
    }
}

2.二叉树的最大宽度

题目链接:

https://leetcode.cn/problems/maximum-width-of-binary-tree/description/

算法思路:

依旧是利⽤层序遍历,但是这⼀次队列⾥⾯不单单存结点信息,并且还存储当前结点如果在数组中存 储所对应的下标(在我们学习数据结构 - 堆的时候,计算左右孩⼦的⽅式)。

这样我们计算每⼀层宽度的时候,⽆需考虑空节点,只需将当层结点的左右结点的下标相减再加 1 即可。

但是,这⾥有个细节问题:如果⼆叉树的层数⾮常恐怖的话,我们任何⼀种数据类型都不能存下下标的值。但是没有问题,因为

• 我们数据的存储是⼀个环形的结构;

• 并且题⽬说明,数据的范围在 int 这个类型的最⼤值的范围之内,因此不会超出⼀圈;

• 因此,如果是求差值的话,我们⽆需考虑溢出的情况。

代码展示:

代码语言:javascript
复制
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public int widthOfBinaryTree(TreeNode root) {
       if(root==null){
        return 0;
       }
       int max=1;
       Queue<Pair<TreeNode,Integer>> q=new LinkedList<>();
       q.add(new Pair(root,1));
       while(!q.isEmpty()){
         int sz=q.size();
         int head=0,last=0;
         for(int i=0;i<sz;i++){
             Pair<TreeNode,Integer> p=q.poll();
             TreeNode cur=p.getKey();
             int v=p.getValue();
             if(cur.left!=null){
                q.add(new Pair(cur.left,v*2));
             }
             if(cur.right!=null){
                q.add(new Pair(cur.right,v*2+1));
             }
            if(i==0){
              head=v;
            }
            if(i==(sz-1)){
                last=v;
            }
         }
         max=max>(last-head+1)?max:(last-head+1);
       }
       return max;
    }
}

3.在每个树行中找最大值

题目链接:

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

算法思路:

层序遍历过程中,在执⾏让下⼀层节点⼊队的时候,我们是可以在循环中统计出当前层结点的最⼤值的。

因此,可以在 bfs 的过程中,统计出每⼀层结点的最⼤值。

代码展示:

代码语言:javascript
复制
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public List<Integer> largestValues(TreeNode root) {
        List<Integer> list=new ArrayList<>();
        if(root==null){
            return list;
        }
        Queue<TreeNode> q=new LinkedList<>();
        q.add(root);
        while(!q.isEmpty()){
            int sz=q.size();
            int max=Integer.MIN_VALUE;
            for(int i=0;i<sz;i++){
                TreeNode cur=q.poll();
                if(cur.left!=null){
                    q.add(cur.left);
                }
                if(cur.right!=null){
                    q.add(cur.right);
                }
                max=max>cur.val?max:cur.val;
            }
            list.add(max);
        }
        return list;
    }
}
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2024-04-07,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • BFS(宽搜)
    • 1.N叉树的层序遍历
      • 2.二叉树的最大宽度
        • 3.在每个树行中找最大值
        相关产品与服务
        对象存储
        对象存储(Cloud Object Storage,COS)是由腾讯云推出的无目录层次结构、无数据格式限制,可容纳海量数据且支持 HTTP/HTTPS 协议访问的分布式存储服务。腾讯云 COS 的存储桶空间无容量上限,无需分区管理,适用于 CDN 数据分发、数据万象处理或大数据计算与分析的数据湖等多种场景。
        领券
        问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档