矩阵距离
给定一个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]=min_{1≤x≤N}{1≤y≤M},A[x][y]=1dist(A[i][j],A[x][y])
B[i][j]=min1≤x≤N1≤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和dfs不同的是我们只会更新一次值。
#include<bits/stdc++.h>
using namespace std;
const int N=1005;
char ma[N][N];
int dis[N][N],m,n;
int dir[4][2]={{1,0},{0,1},{-1,0},{0,-1}};
queue<pair<int,int> >q;
void bfs()
{
while(!q.empty()) q.pop();
for(int i=0;i<n;i++){
for(int j=0;j<m;j++){
if(ma[i][j]=='1'){
dis[i][j]=0;
q.push({i,j});
}
}
}
while(!q.empty()){
pair<int,int> u=q.front();q.pop();
for(int i=0;i<4;i++){
int x=u.first+dir[i][0];
int y=u.second+dir[i][1];
if(x<0||x>=n||y<0||y>=m) continue;
if(dis[x][y]==-1){
dis[x][y]=dis[u.first][u.second]+1;
q.push({x,y});
}
}
}
}
int main()
{
cin>>n>>m;
for(int i=0;i<n;i++) cin>>ma[i];
memset(dis,-1,sizeof dis);
bfs();
for(int i=0;i<n;i++){
for(int j=0;j<m;j++){
printf("%d ",dis[i][j]);
}
puts("");
}
}