使用队列进行BFS
#include <iostream>
#include <cstring>
#include <cmath>
#include <cstdio>
#include <algorithm>
#include <queue>
using namespace std;
typedef pair<int, int> PII;
const int N = 1010;
int n, m;
char g[N][N]; //存放输入的01矩阵
int cnt[N][N]; //存放输出的整数矩阵
int dx[4] = {0, -1, 0, 1}, dy[4] = {-1, 0, 1, 0};
queue<PII> q;
int main()
{
memset(cnt, -1, sizeof cnt);
cin >> n >> m;
for(int i=0; i<n; i++)
cin >> g[i];
for(int i=0; i<n; i++)
for(int j=0; j<m; j++)
if(g[i][j] == '1')
{
cnt[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.first + dx[i], y = t.second + dy[i];
if(x>=0 && x<n && y>=0 && y<m && cnt[x][y] == -1)
{
cnt[x][y] = cnt[t.first][t.second] + 1;
q.push({x, y});
}
}
}
for(int i=0; i<n; i++)
{
for(int j=0; j<m; j++)
cout << cnt[i][j] << ' ';
cout << endl;
}
return 0;
}