题目: AcWing 173. 矩阵距离
给定一个 N 行 M 列的 01 矩阵 A,A[i][j] 与 A[k][l] 之间的曼哈顿距离定义为:
输出一个 N 行 M 列的整数矩阵 B,其中:
输入格式
第一行两个整数 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
题目分析:
这是一个多源搜索的问题,但是我们可以简化为普通的广度优先搜索。
根据示例我们可以知道,点1的距离为0,如果我们根据将点1全部压入队列中,使用BFS进行搜索,可以得到如下结果,最后这就是我们想要的。
为什么该题目多源搜索,可以转化为简单广度优先搜索?
我们可以理解为从一个源点到达1的点的距离为0,这样更好理解。
#include <iostream>
#include <queue>
#include <cstring>
#include <algorithm>
using namespace std;
typedef pair<int,int> PII;
const int N = 1010;
char c[N][N];
int d[N][N];
int dx[]={-1,0,1,0};
int dy[]={0,1,0,-1};
int n,m;
void bfs()
{
memset(d,-1,sizeof(d));
queue<PII> q;
for(int i=0;i<n;i++) // 将为1的点加入队列
for(int j=0;j<m;j++)if(c[i][j]=='1')q.push({i,j}),d[i][j]=0;
while(!q.empty())
{
auto s=q.front();
int x=s.first,y=s.second;
for(int i=0;i<4;i++)
{
int a=x+dx[i],b=y+dy[i];
if(a<0||b<0||a>=n||b>=m||d[a][b]!=-1)continue;
q.push({a,b});
d[a][b]=d[x][y]+1;
}
q.pop();
}
}
int main()
{
cin>>n>>m;
for(int i=0;i<n;i++)cin>>c[i];
bfs();
for(int i=0;i<n;i++)
{
for(int j=0;j<m;j++) cout<<d[i][j]<<' ';
cout<<endl;
}
return 0;
}