AcWing算法提高课-2.3.1矩阵距离

算法提高课整理

CSDN个人主页:更好的阅读体验


本文同步发表于 CSDN | 洛谷 | AcWing | 个人博客

原题链接
题目描述

给定一个 01 矩阵,求矩阵中每个元素离 1 的最短曼哈顿距离。

输入格式

第一行两个整数 n , m n,m n,m。

接下来一个 n n n 行 m m m 列的 01 矩阵,数字之间没有空格。

输出格式

一个 n n n 行 m m m 列的矩阵,相邻数字之间用空格隔开。

数据范围

1 ≤ n , m ≤ 1000 1\le n,m\le 1000 1≤n,m≤1000


思路

先考虑从 0 的位置向外扩展。

发现这样的话较麻烦,于是改为考虑从 1 的位置用 BFS 向外扩展,并处理出所有的距离。

这种算法即为 "多源 BFS"。具体算法流程为:将所有源点都入队,然后正常跑 BFS。

具体细节见代码。

算法时间复杂度
AC Code

C + + \text{C}++ C++

cpp 复制代码
#include <iostream>
#include <cstring>
#include <queue>

using namespace std;

typedef pair<int, int> PII;
#define x first
#define y second

const int N = 1010;
int dx[] = {-1, 0, 1, 0};
int dy[] = {0, 1, 0, -1};

int n, m;
char g[N][N];
int dist[N][N];

void bfs()
{
    memset(dist, -1, sizeof dist);
    queue<PII> q;
    
    for (int i = 0; i < n; i ++ )
        for (int j = 0; j < m; j ++ )
            if (g[i][j] == '1')
                dist[i][j] = 0, q.push({i, j}); // 所有起点入队
    
    while (q.size())
    {
        PII t = q.front();
        q.pop();
        
        for (int i = 0; i < 4; i ++ ) // 4方向扩展
        {
            int x = t.x + dx[i], y = t.y + dy[i];
            if (x < 0 || x >= n || y < 0 || y >= m) continue; // 出界
            if (dist[x][y] != -1) continue; // 已经被遍历过
            dist[x][y] = dist[t.x][t.y] + 1; // 合法的话更新距离
            q.push({x, y}); // 新点入队
        }
    }
}

int main()
{
    scanf("%d%d", &n, &m);
    for (int i = 0; i < n; i ++ )
        scanf("%s", g[i]);
    
    bfs();
    
    for (int i = 0; i < n; i ++ )
    {
        for (int j = 0; j < m; j ++ )
            printf("%d ", dist[i][j]);
        puts("");
    }
    
    return 0;
}

最后,如果觉得对您有帮助的话,点个赞再走吧!

相关推荐
小小小小王王王5 分钟前
求猪肉价格最大值
数据结构·c++·算法
岁忧29 分钟前
(LeetCode 面试经典 150 题 ) 58. 最后一个单词的长度 (字符串)
java·c++·算法·leetcode·面试·go
BIYing_Aurora37 分钟前
【IPMV】图像处理与机器视觉:Lec13 Robust Estimation with RANSAC
图像处理·人工智能·算法·计算机视觉
码农编程录2 小时前
【c/c++3】类和对象,vector容器,类继承和多态,systemd,std&boost
c++
martian6652 小时前
支持向量机(SVM)深度解析:从数学根基到工程实践
算法·机器学习·支持向量机
孟大本事要学习2 小时前
算法19天|回溯算法:理论基础、组合、组合总和Ⅲ、电话号码的字母组合
算法
??tobenewyorker3 小时前
力扣打卡第二十一天 中后遍历+中前遍历 构造二叉树
数据结构·c++·算法·leetcode
让我们一起加油好吗3 小时前
【基础算法】贪心 (二) :推公式
数据结构·数学·算法·贪心算法·洛谷
贾全3 小时前
第十章:HIL-SERL 真实机器人训练实战
人工智能·深度学习·算法·机器学习·机器人
GIS小天4 小时前
AI+预测3D新模型百十个定位预测+胆码预测+去和尾2025年7月4日第128弹
人工智能·算法·机器学习·彩票