给定一个 NN 行 MM 列的 0101 矩阵 AA,A[i][j]A[i][j] 与 A[k][l]A[k][l] 之间的曼哈顿距离定义为:
dist(A[i][j],A[k][l])=|i−k|+|j−l|dist(A[i][j],A[k][l])=|i−k|+|j−l|
输出一个 NN 行 MM 列的整数矩阵 BB,其中:
B[i][j]=min1≤x≤N,1≤y≤M,A[x][y]=1dist(A[i][j],A[x][y])B[i][j]=min1≤x≤N,1≤y≤M,A[x][y]=1dist(A[i][j],A[x][y])
输入格式
第一行两个整数 N,MN,M。
接下来一个 NN 行 MM 列的 0101 矩阵,数字之间没有空格。
输出格式
一个 NN 行 MM 列的矩阵 BB,相邻两个整数之间用一个空格隔开。
数据范围
1≤N,M≤10001≤N,M≤1000
输入样例:
3 4 0001 0011 0110
输出样例:
3 2 1 0 2 1 0 0 1 0 0 1
/*
将矩阵中所有为值为1的读入队列;从1开始搜索,每个不是1的点如果第一次被搜到则必为距离1最近的点
*/
#include <queue>
#include <cstdio>
#include <cstring>
#include <iostream>
#define x first
#define y second
using namespace std;
typedef pair<int, int> PII;
const int N = 1010;
int n, m;
char g[N][N];
int dist[N][N];
int dx[4] = {-1, 1, 0, 0};
int dy[4] = {0, 0, -1, 1};
void bfs()
{
queue<PII> q;
memset(dist, -1, sizeof dist);
for (int i = 0; i < n; i ++ )//将所有数值为1的点读入数列
for (int j = 0; j < m; j ++ )
if (g[i][j] == '1')
{
dist[i][j] = 0;
q.push({i, j});
}
while (q.size())
{
auto t = q.front();
q.pop();
for (int i = 0; i < 4; i ++ )
{
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()
{
cin >> 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;
}