题目描述
给定一个 N 行 M 列的 01 矩阵 A,A[i][j] 与 A[k][l] 之间的曼哈顿距离定义为:
dist(A[i][j],A[k][l])=|i−k|+|j−l|
输出一个 N 行 M 列的整数矩阵 B,其中:
B[i][j]=min1≤x≤N,1≤y≤M,A[x][y]=1dist(A[i][j],A[x][y])
输入格式
第一行两个整数 N,M。
接下来一个 N 行 M 列的 01 矩阵,数字之间没有空格。
输出格式
一个 N 行 M 列的矩阵 B,相邻两个整数之间用一个空格隔开。
数据范围
1≤N,M≤1000
样例
输入样例:
3 4
0001
0011
0110
输出样例:
3 2 1 0
2 1 0 0
1 0 0 1
C++ 代码
#include<bits/stdc++.h>
#include<queue>
using namespace std;
int n,m;
char g[1020][1020];
int dist[1020][1020];
int dx[4] = {0,1,0,-1},dy[4] = {-1,0,1,0};
struct node{
int x,y;
};
void bfs(){
queue<node> q;
memset(dist,-1,sizeof dist);
for(int i = 0; i < n; i++){
for(int j = 0; j < m; j++){
if(g[i][j]=='1'){
q.push({i,j});
dist[i][j] = 0;
}
}
}
while(q.size()){
node 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(g[x][y]=='1'||dist[x][y]!=-1) continue;
g[x][y]='1';
q.push({x,y});
dist[x][y] = dist[t.x][t.y] + 1;
}
}
}
int main(){
scanf("%d%d",&n,&m);
for(int i = 0; i < n;i++){
cin>>g[i];
}
bfs();
for(int i = 0; i < n;i++){
for(int j = 0; j < m;j++){
printf("%d ",dist[i][j]);
}cout<<endl;
}
return 0;
}