[算法题]01 矩阵

题目链接: 01 矩阵

多源BFS类型题, 即给定多个起点, 判断从哪个起点走到终点距离最短, 一般解题思路为将所有起点看成一个"起点", 由此"起点"做bfs得到题解, 实际代码编写将所有起点都入队列, 每次都对所有起点做一层扩展.

题解思路:

从1往0处走寻找最短距离不好操作, 逆向思维从0往1处走, 这样很好更新最短距离, 如图:

第一层的值为0, 那么扩展后的值就是0+1

第二层的值为1, 那么扩展后的值就是1+1

题解大致步骤:

  1. 定义方向数组.

  2. 定义一个与mat同等规模的二维数组, 将其初始值设为-1, 当数组对应坐标的值为-1时表示当前位置未被访问过, 当数组对应坐标的值不为-1时则表示该坐标到0的最短距离.

  3. 定义一个存储pair<int,int>的队列, 并将mat数组中0的坐标入队, 同时将步骤2定义的数组的同等坐标位置的值置为0.

  4. 做一次多源bfs, 每次取队头的值向四周扩一次, 并根据条件更新队列和步骤2的数组, 最后步骤2的数组就是题解, 返回即可.

题解代码:

cpp 复制代码
class Solution 
{
public:
    vector<vector<int>> updateMatrix(vector<vector<int>>& mat) 
    {
        //方向数组
        int dx[4] = {0, 0, -1, 1};
        int dy[4] = {1, -1, 0, 0};

        int row = mat.size();
        int col = mat[0].size();
        vector<vector<int>> res(row, vector<int>(col, -1));
        queue<pair<int,int>> q;

        //添加0的坐标到q
        for(int i = 0; i < row; ++i)
        {
            for(int j = 0; j < col; ++j)
            {
                if(mat[i][j] == 0)
                {
                    q.push({i, j});
                    res[i][j] = 0; //同时修改结果数组
                }
            }
        }

        //多源bfs
        while(q.size())
        {
            auto [a, b] = q.front();
            q.pop();
            for(int j = 0; j < 4; ++j)
            {
                int x = a + dx[j];
                int y = b + dy[j];
                if(x >= 0 && x < row && y >= 0 && y < col && res[x][y] == -1)
                {
                    q.push({x, y});
                    res[x][y] = res[a][b] + 1;
                }
            }
        }

        return res;
    }
};
相关推荐
??tobenewyorker42 分钟前
力扣打卡第二十一天 中后遍历+中前遍历 构造二叉树
数据结构·c++·算法·leetcode
贾全1 小时前
第十章:HIL-SERL 真实机器人训练实战
人工智能·深度学习·算法·机器学习·机器人
GIS小天1 小时前
AI+预测3D新模型百十个定位预测+胆码预测+去和尾2025年7月4日第128弹
人工智能·算法·机器学习·彩票
满分观察网友z2 小时前
开发者的“右”眼:一个树问题如何拯救我的UI设计(199. 二叉树的右视图)
算法
森焱森3 小时前
无人机三轴稳定化控制(1)____飞机的稳定控制逻辑
c语言·单片机·算法·无人机
循环过三天3 小时前
3-1 PID算法改进(积分部分)
笔记·stm32·单片机·学习·算法·pid
闪电麦坤953 小时前
数据结构:二维数组(2D Arrays)
数据结构·算法
凌肖战4 小时前
力扣网C语言编程题:快慢指针来解决 “寻找重复数”
c语言·算法·leetcode
埃菲尔铁塔_CV算法4 小时前
基于 TOF 图像高频信息恢复 RGB 图像的原理、应用与实现
人工智能·深度学习·数码相机·算法·目标检测·计算机视觉
NAGNIP5 小时前
一文搞懂FlashAttention怎么提升速度的?
人工智能·算法