多源BFS问题(1)_01矩阵

个人主页:C++忠实粉丝
欢迎 点赞👍 收藏✨ 留言✉ 加关注💓本文由 C++忠实粉丝 原创

多源BFS问题(1)_01矩阵

收录于专栏【经典算法练习
本专栏旨在分享学习算法的一点学习笔记,欢迎大家在评论区交流讨论💌

目录

[1. 题目链接](#1. 题目链接)

[2. 题目描述](#2. 题目描述)

[3. 解法](#3. 解法)

算法思路:

代码展示:

[4. 解法](#4. 解法)


1. 题目链接

OJ链接 : 01矩阵

2. 题目描述

给定一个由 01 组成的矩阵 mat ,请输出一个大小相同的矩阵,其中每一个格子是 mat 中对应位置元素到最近的 0 的距离。

两个相邻元素间的距离为 1

示例 1:

复制代码
输入:mat = [[0,0,0],[0,1,0],[0,0,0]]
输出:[[0,0,0],[0,1,0],[0,0,0]]

示例 2:

复制代码
输入:mat = [[0,0,0],[0,1,0],[1,1,1]]
输出:[[0,0,0],[0,1,0],[1,2,1]]

提示:

  • m == mat.length
  • n == mat[i].length
  • 1 <= m, n <= 104
  • 1 <= m * n <= 104
  • mat[i][j] is either 0 or 1.
  • mat 中至少有一个 0

3. 解法

算法思路:

对于求的最终结果, 我们有两种方式 :

第一种方式 : 从每一个1开始, 然后通过层序遍历找到离它最近的0

这第一种方式, 我们会以所有的1起点, 来一次层序遍历, 势必会遍历很多重复的点. 并且如果矩阵中只有一个0的话, 每一次层序遍历都要遍历很多层, 时间复杂度较高.

第二种方式 : 从0开始层序遍历, 并且记录遍历的层数, 当第一次碰到1的时候, 当前的层数就是这个1离0的最短距离.

第二种方式, 我们在遍历的时候标记一下处理过的1, 能够做到只用遍历整个矩阵一次, 就能得到最终结果

但是, 这里有一个问题, 0是有很多个的, 我们怎么才能保证遇到的1距离这一个0是最近的呢?

其实很简单, 我们可以先把所有的0都放在队列种, 把它们当成一个整体, 每次当前队列里面的所有元素向外扩展一次.

如下图所示:

代码展示:

cpp 复制代码
class Solution {
    int dx[4] = {0, 0, 1, -1};
    int dy[4] = {1, -1, 0, 0};
public:
    vector<vector<int>> updateMatrix(vector<vector<int>>& mat) 
    {
        int n = mat.size(), m = mat[0].size();

        //dist[i][j] = -1 表示: 没有搜索过
        //dist[i][j] != -1表示: 最短距离
        vector<vector<int>> dist(n, vector<int>(m, -1));
        queue<pair<int, int>> q;

        //1. 把所有的源点加入到队列中
        for(int i = 0; i < n; i++)
            for(int j = 0; j < m; j++)
                if(mat[i][j] == 0)
                {
                    q.push({i, j});
                    dist[i][j] = 0;
                }

        //2. 一层一层的往外扩
        while(q.size())
        {
            auto [a, b] = q.front();
            q.pop();
            for(int i = 0; i < 4; i++)
            {
                int x = a + dx[i], y = b + dy[i];
                if(x >= 0 && x < n && y >= 0 && y < m && dist[x][y] == -1)
                {
                    dist[x][y] = dist[a][b] + 1;
                    q.push({x, y});
                }
            }
        }
        return dist;
    }
};

4. 算法总结

总结
时间复杂度:O(n* m) ,其中 n 和 m 是矩阵的行和列数,因为每个元素最多被访问一次。
空间复杂度:O(n* m),用于存储距离矩阵和队列。

相关推荐
自动驾驶小卡1 小时前
boost::circular_buffer的使用方法简介
c++·boost·circular_buffer
Greedy Alg3 小时前
LeetCode 142. 环形链表 II
算法
睡不醒的kun3 小时前
leetcode算法刷题的第三十二天
数据结构·c++·算法·leetcode·职场和发展·贪心算法·动态规划
先做个垃圾出来………4 小时前
残差连接的概念与作用
人工智能·算法·机器学习·语言模型·自然语言处理
乔宕一5 小时前
stm32 链接脚本没有 .gcc_except_table 段也能支持 C++ 异常
c++·stm32·嵌入式硬件
SuperCandyXu5 小时前
P3205 [HNOI2010] 合唱队-普及+/提高
c++·算法·洛谷
_OP_CHEN6 小时前
数据结构(C语言篇):(十二)实现顺序结构二叉树——堆
c语言·数据结构·算法·二叉树·学习笔记··顺序结构二叉树
_君落羽_6 小时前
ARM寄存器以及异常处理
c++
Yingjun Mo6 小时前
1. 统计推断-基于神经网络与Langevin扩散的自适应潜变量建模与优化
人工智能·神经网络·算法·机器学习·概率论
free6 小时前
基于librdkafa C++客户端生产者发送数据失败问题处理#2
c++·kafka