《算法竞赛进阶指南》 矩阵距离

给定一个 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]=1⁡dist(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;
}