距离矩阵
可以发现哪个公式就是距离最近的1的距离
我们利用bfs先遍历到的点一定最短这个性质,建立超级源点,将所有的1入队,那么这些1会先进行bfs遇见0,那么这个0一定是最先看见当前的1,只需要更新一下距离即可
#include <bits/stdc++.h>
#define ll long long
#define x first
#define y second
#define INF 0x7f7f7f7f
using namespace std;
typedef pair<int,int> PII;
const int N=1e3+10;
int n,m,ans;
int dx[4]= {1,0,-1,0};
int dy[4]= {0,1,0,-1};
int vis[N][N];
string g[N];
int main() {
cin>>n>>m;
for(int i=0; i<n; i++)cin>>g[i];
queue<PII> q;
//超级源点,将所有的1放入队列
memset(vis,INF,sizeof(vis));
for(int i=0; i<n; i++)
for(int j=0; j<m; j++) {
if(g[i][j]=='1') {
q.push({i,j});
vis[i][j]=0;
}
}
while(!q.empty()) {
PII p=q.front();
q.pop();
for(int i=0; i<4; i++) {
int nx=p.x+dx[i];
int ny=p.y+dy[i];
if(nx<0||ny<0||nx>=n||ny>=m||vis[nx][ny]!=INF)continue;
vis[nx][ny]=vis[p.x][p.y]+1;
q.push({nx,ny});
}
}
for(int i=0;i<n;i++){
for(int j=0;j<m;j++){
cout<<vis[i][j]<<" ";
}
cout<<endl;
}
return 0;
}