-
4 4 0110 1111 1111 0110
样例输出
-
0 1 1 0 1 2 2 1 1 2 2 1 0 1 1 0
描述
给定一个N x M的01矩阵,其中1表示陆地,0表示水域。对于每一个位置,求出它距离最近的水域的距离是多少。
矩阵中每个位置与它上下左右相邻的格子距离为1。
输入
第一行包含两个整数,N和M。
以下N行每行M个0或者1,代表地图。
数据保证至少有1块水域。
对于30%的数据,1 <= N, M <= 100
对于100%的数据,1 <= N, M <= 800
输出
输出N行,每行M个空格分隔的整数。每个整数表示该位置距离最近的水域的距离。
从水域出发进行搜索,将所有水域坐标放入队列,然后进行spfa算法操作,求出到水域最短距离。
我这里没有标记是否进入队列,实际上也就退化成了BellmanFord算法,但是这里点与相邻点距离都是1,
挪动次数等于走的距离,每个点只进入队列一次,所以时间复杂度并没有影响。。
#include <bits/stdc++.h>
using namespace std;
const int maxn=1e5+7;
typedef long long ll;
char g[1000][1000];
int n,m;
int dir[4][2]={{0,1},{1,0},{0,-1},{-1,0}};
int ans[1000][1000];
int main()
{
scanf("%d%d",&n,&m);
for(int i=0;i<n;i++){
scanf("%s",g[i]);
}
for(int i=0;i<n;i++){
fill(ans[i],ans[i]+m+3,10000000);
}
queue<pair<int,int > >que;
for(int i=0;i<n;i++){
for(int j=0;j<m;j++){
if(g[i][j]=='0'){ans[i][j]=0;que.push(make_pair(i,j));}
}
}
while(!que.empty()){
pair<int,int > p=que.front();
que.pop();
for(int i=0;i<4;i++){
int xx=p.first+dir[i][0];
int yy=p.second+dir[i][1];
if(xx>=0&&xx<n&&yy>=0&&yy<m&&ans[p.first][p.second]+1<ans[xx][yy]){
ans[xx][yy]=ans[p.first][p.second]+1;
que.push(make_pair(xx,yy));
}
}
}
for(int i=0;i<n;i++){
for(int j=0;j<m;j++){
printf("%d",ans[i][j]);
if(j!=m-1)printf(" ");
}
printf("\n");
}
return 0;
}