算法提高课整理

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

---

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

原题链接

题目描述

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

输入格式

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

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

输出格式

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

数据范围

$1\le n,m\le 1000$

---

思路

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

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

这种算法即为 “多源 BFS”。具体算法流程为：将所有源点都入队，然后正常跑 BFS。

具体细节见代码。

算法时间复杂度

AC Code

$\text{C}++$

#include <iostream>
#include <cstring>
#include <queue>using namespace std;typedef pair<int, int> PII;
#define x first
#define y secondconst 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;
}

---

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