AcWing 173. 矩阵距离
原题链接
简单
作者:
渐修
,
2024-04-12 20:30:02
,
所有人可见
,
阅读 5
/*
多起点最短路径,把所有起点都加入队列进行bfs
*/
#include<iostream>
#include<queue>
#include<cstring>
#define x first
#define y second
using namespace std;
typedef pair<int,int> pai;
const int N=1010;
int n,m;
int d[N][N];
int dx[4]={0,0,-1,1},dy[4]={1,-1,0,0};
char g[N][N];
queue<pai> q;
void bfs()
{
for(int i=0;i<n;i++)
for(int j=0;j<m;j++)
if(g[i][j]=='1')
{
d[i][j]=0;
q.push({i,j});
}
while(q.size())
{
pai t=q.front();
q.pop();
for(int i=0;i<4;i++)
{
int x=dx[i]+t.x,y=dy[i]+t.y;
if(x<0||x>=n||y<0||y>=m)
continue;
if(d[x][y]!=-1)
continue;
d[x][y]=d[t.x][t.y]+1;
q.push({x,y});
}
}
}
int main()
{
memset(d,-1,sizeof d);
cin>>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++)
cout<<d[i][j]<<" ";
puts("");
}
return 0;
}