水陆距离 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

题意：

记录每个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;
}