水陆距离 HihoCoder - 1478

 水陆距离

  HihoCoder - 1478

给定一个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

代码：

#include<stdio.h>
#include<string.h>
#include<queue>
#include<algorithm>
using namespace std;
int n,m,go[4][2]={0,1,0,-1,1,0,-1,0},d[805][805];
char map[805][805];
struct node
{
    int x,y,s;
}no,ne,t;
queue<node>q;
void bfs()
{
    while(!q.empty())
    {
        no=q.front();
        q.pop();
        for(int i=0;i<4;i++)
        {
            int xx=no.x+go[i][0];
            int yy=no.y+go[i][1];
            if(xx>=0&&xx<n&&yy>=0&&yy<m&&d[xx][yy]==-1)
            {
                ne.s=no.s+1;
                ne.x=xx,ne.y=yy;
                d[xx][yy]=ne.s;
                q.push(ne);
            }
        }
    }
}
int main()
{
    memset(d,-1,sizeof(d));
   scanf("%d%d",&n,&m);
   for(int i=0;i<n;i++)
        scanf("%s",map[i]);
   for(int i=0;i<n;i++)
    for(int j=0;j<m;j++)
   {
       if(map[i][j]=='0')
       {
           d[i][j]=0;
           t.x=i,t.y=j,t.s=0;
           q.push(t);
       }
   }
   bfs();
   for(int i=0;i<n;i++)
   {
       for(int j=0;j<m;j++)
        printf("%d ",d[i][j]);
       printf("\n");
   }
   return 0;
}