AcWing 3083. 矩阵距离(两个矩阵距离代码一模一样)
原题链接
简单
要求每个0到1的最近曼哈顿距离,实际上就是以1为源头
进行多源BFS,把整个数组跑一遍就好了
#include <iostream>
#include <cstring>
using namespace std;
typedef pair<int,int> PII;
char g[1010][1010];
int dist[1010][1010];
int hh=0,tt=-1;
PII q[1010*1010];
int dx[]={-1,0,1,0};
int dy[]={0,1,0,-1};
int n,m;
void bfs(){
while (hh<=tt){
auto t=q[hh++];
for(int i=0;i<4;i++){
int a=t.first+dx[i];
int b=t.second+dy[i];
if (a<1||a>n||b<1||b>m) continue;
if (dist[a][b]>=0) continue;
q[++tt]={a,b};
dist[a][b]=dist[t.first][t.second]+1;
}
}
}
int main(){
cin>>n>>m;
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
cin>>g[i][j];
dist[i][j]=g[i][j]-'0';
}
}
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
if (dist[i][j]==1){
q[++tt]={i,j};
dist[i][j]=0;
}
else{
dist[i][j]=-1;
}
}
}
bfs();
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
printf("%d ",dist[i][j]);
}
printf("\n");
}
return 0;
}