AcWing 173. 矩阵距离
给定一个N行M列的01矩阵A,A[i][j] 与 A[k][l] 之间的曼哈顿距离定义为:
d
i
s
t
(
A
[
i
]
[
j
]
,
A
[
k
]
[
l
]
)
=
∣
i
−
k
∣
+
∣
j
−
l
∣
dist(A[i][j],A[k][l])=|i−k|+|j−l|
dist(A[i][j],A[k][l])=∣i−k∣+∣j−l∣
输出一个N行M列的整数矩阵B,其中:
B
[
i
]
[
j
]
=
m
i
n
1
≤
x
≤
N
,
1
≤
y
≤
M
,
A
[
x
]
[
y
]
=
1
d
i
s
t
(
A
[
i
]
[
j
]
,
A
[
x
]
[
y
]
)
B[i][j]=min1≤x≤N,1≤y≤M,A[x][y]=1dist(A[i][j],A[x][y])
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
这道题是多源bfs。
y总的理解我有点不是很明白。
我的理解是,以目标点为中心,然后扫描过里中心点的距离就是最短路程。
那我们就按bfs的模板,把每一个中心点插进去,然后bfs,因为bfs队列的特殊性质,所以每个目标点都以此为中心,每次遍历一个中心点扩散,然后循环扩散。
代码如下
#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
#define x first
#define y second
using namespace std;
typedef pair<int,int> PII;
const int N=1010;
int dist[N][N];
bool st[N][N];
int n,m;
queue<PII> q;
void bfs()
{
int dx[]={0,0,1,-1},dy[]={1,-1,0,0};
while(q.size())
{
auto t=q.front();
q.pop();
for(int i=0;i<4;i++)
{
int tx=t.x+dx[i];
int ty=t.y+dy[i];
if(tx<1||ty<1||tx>n||ty>m)
continue;
if(!st[tx][ty])
{
st[tx][ty]=true;
dist[tx][ty]=min(dist[tx][ty],dist[t.x][t.y]+1);
q.push({tx,ty});
}
}
}
}
int main(void)
{
memset(dist,0x3f,sizeof dist);
scanf("%d %d",&n,&m);
// exit(0);
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
int t;
scanf("%1d",&t);
if(t)
{
dist[i][j]=0;
q.push({i,j});
st[i][j]=true;
}
}
bfs();
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
printf("%d ",dist[i][j]);
puts("");
}
}