【ACWing】3083. 矩阵距离

题目地址：

https://www.acwing.com/problem/content/3086/

给定一个$n\times m$的$01$矩阵：

a11 a12 ... a1m
...
an1 an2 ... anm

定义$a_{ij}$与$a_{kl}$之间的距离为$D(a_{ij},a_{kl})=|i-k|+|j-l|$。
对于每个元素$a_{ij}$，请求出与它距离最近且值为$1$的元素$a_{kl}$和它的距离是多少。
另外注意，当元素$a_{ij}$本身就为$1$时，与它距离最近且值为$1$的元素就是它自己，距离为$0$。

输入格式：
第一行为两个整数，分别代表$n$和$m$。
接下来的$n$行，第$i$行的第$j$个字符代表$a_{ij}$。

输出格式：
共$n$行，第$i$行的第$j$个数字表示$a_{ij}$与其距离最近且值为$1$的元素的距离。

数据范围：
$1\le n,m\le 1000$

多源BFS，从所有的$1$出发做BFS即可。代码如下：

#include <cstring>
#include <iostream>
#include <queue>
using namespace std;

const int N = 1010;
int m, n;
char s[N][N];
int res[N][N];
queue<pair<int, int>> q;

void bfs() {
  memset(res, -1, sizeof res);
  for (int i = 1; i <= m; i++)
    for (int j = 1; j <= n; j++)
      if (s[i][j] == '1') {
        q.push({i, j});
        res[i][j] = 0;
      }

  static int d[] = {-1, 0, 1, 0, -1};
  while (q.size()) {
    auto p = q.front();
    q.pop();
    int x = p.first, y = p.second;
    for (int k = 0; k < 4; k++) {
      int nx = x + d[k], ny = y + d[k + 1];
      if (1 <= nx && nx <= m && 1 <= ny && ny <= n && !~res[nx][ny]) {
        res[nx][ny] = res[x][y] + 1;
        q.push({nx, ny});
      }
    }
  }
}

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

时空复杂度$O(mn)$。