给定一个N x M的01矩阵,其中1表示陆地,0表示水域。对于每一个位置,求出它距离最近的水域的距离是多少。
矩阵中每个位置与它上下左右相邻的格子距离为1。
Input
第一行包含两个整数,N和M。
以下N行每行M个0或者1,代表地图。
数据保证至少有1块水域。
对于30%的数据,1 <= N, M <= 100
对于100%的数据,1 <= N, M <= 800
Output
输出N行,每行M个空格分隔的整数。每个整数表示该位置距离最近的水域的距离。
Sample Input
4 4
0110
1111
1111
0110
Sample Output
0 1 1 0
1 2 2 1
1 2 2 1
0 1 1 0
题意:
记录每个1到0的最小距离,以图的形式输出
思路:
对四个方向进行广度搜索,每一次搜索1到0,走过的路进行标记,记录走的最小步数
代码:
#include<stdio.h>
#include<queue>
#include<string.h>
using namespace std;
int n,m;
char a[810][810];
int b[810][810],book[810][810];
int dir[4][2]= {0,1,0,-1,1,0,-1,0};
struct water
{
int x;
int y;
int s;
} u,v;
queue<water>q;
int bfs()
{
int i,tx,ty;
while(!q.empty())
{
u=q.front();
q.pop();
for(i=0; i<4; i++)
{
tx=u.x+dir[i][0];
ty=u.y+dir[i][1];
if(tx>=0&&tx<n&&ty>=0&&ty<m&&a[tx][ty]=='1'&&book[tx][ty]==0)
{
book[tx][ty]=1;
v.x=tx;
v.y=ty;
v.s=u.s+1;
b[tx][ty]=v.s;
q.push(v);
}
}
}
}
int main()
{
scanf("%d %d",&n,&m);
int i,j;
for(i=0; i<n; i++)
scanf("%s",a[i]);
for(i=0; i<n; i++)
{
for(j=0; j<m; j++)
{
if(a[i][j]=='0')
{
b[i][j]=0;
book[i][j]=1;
u.x=i;
u.y=j;
u.s=0;
q.push(u);
}
}
}
bfs();
for(i=0; i<n; i++)
{
for(j=0; j<m; j++)
{
if(j==0)
printf("%d",b[i][j]);
else
printf(" %d",b[i][j]);
}
printf("\n");
}
return 0;
}