题意:
思路:将所有位置为1的点全部入队,然后进行BFS,保证每个点只被搜索一次,时间复杂度O(n*m)。在对每个点往外扩的时候,它能够到达的点一定是距当前点最近的,所以一定能够保证每个位置不为1的点到图中位置为1的点的曼哈顿距离最小。
#include<iostream>
#include<queue>
#include<cstring>
using namespace std;
const int N=1e3+10,INF=0x3f3f3f3f;
int n,m;
int d[N][N];
int dx[]={0,0,1,-1},dy[]={1,-1,0,0};
struct node
{
int x,y;
int s;
};
queue<node> q;
void bfs()
{
while(!q.empty())
{
node t=q.front();
q.pop();
int x=t.x,y=t.y,s=t.s;
d[x][y]=s;
for(int i=0;i<4;i++)
{
int tx=x+dx[i],ty=y+dy[i];
if(tx>=0&&tx<n&&ty>=0&&ty<m)
{
if(d[tx][ty]>s+1)
{
d[tx][ty]=s+1;
q.push({tx,ty,s+1});
}
}
}
}
}
int main()
{
scanf("%d%d",&n,&m);
int x;
memset(d,INF,sizeof(d));
for(int i=0;i<n;i++)
for(int j=0;j<m;j++)
{
scanf("%1d",&x);
if(x==1)
{
q.push({i,j,0});
d[i][j]=0;
}
}
bfs();
for(int i=0;i<n;i++)
for(int j=0;j<m;j++)
{
if(j==m-1)
printf("%d\n",d[i][j]);
else
printf("%d ",d[i][j]);
}
return 0;
}