HihoCoder - 1478 水路距离 【BFS】

题意：给定一个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个空格分隔的整数。每个整数表示该位置距离最近的水域的距离。

思路：首先遍历整个地图，将水域点全加入队列种，然后BFS

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<string>
#include<algorithm>
#include<map>
#include<vector>
#include<queue>
#include<stack>
#include<string>
#include<math.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const double pi=acos(-1.0);
const int N=805;

int ans[N][N];
char mp[N][N];
int vis[N][N];
int n,m;
int d[4][2]= {0,1,1,0,0,-1,-1,0};

struct node
{
    int x,y,s;
};

int judge(int x,int y)//判断水域四周是否有陆地
{
    if(x+1<n&&mp[x+1][y]=='1') return 1;
    if(x-1>=0&&mp[x-1][y]=='1') return 1;
    if(y+1<m&&mp[x][y+1]=='1') return 1;
    if(y-1>=0&&mp[x][y-1]=='1') return 1;
    return 0;
}

queue<node>que;
node u,v;

void bfs()
{
    memset(vis,0,sizeof(vis));
    while(!que.empty())
    {
        u=que.front();
        que.pop();
        for(int i=0; i<4; i++)
        {
            v.x=u.x+d[i][0];
            v.y=u.y+d[i][1];
            if(v.x>=0&&v.x<n&&v.y>=0&&v.y<m&&vis[v.x][v.y]==0&&mp[v.x][v.y]=='1'&&u.s+1<=ans[v.x][v.y])
            {
                vis[v.x][v.y]=1;
                v.s=u.s+1;
                ans[v.x][v.y]=min(ans[v.x][v.y],v.s);
                que.push(v);
            }
        }
    }
}

int main()
{
    scanf("%d %d",&n,&m);
    for(int i=0; i<n; i++)
    {
        for(int j=0; j<m; j++)
            ans[i][j]=inf;
    }
    for(int i=0; i<n; i++)
        scanf("%s",mp[i]);
    for(int i=0; i<n; i++)
    {
        for(int j=0; j<m; j++)
        {
            if(mp[i][j]=='0'&&judge(i,j))
            {
                u.x=i;
                u.y=j;
                u.s=0;
                que.push(u);
            }
        }
    }
    bfs();
    for(int i=0; i<n; i++)
    {
        for(int j=0; j<m; j++)
            if(mp[i][j]=='1') printf("%d ",ans[i][j]);
            else printf("0 ");
        printf("\n");
    }
    return 0;
}