hihocoder1478 水陆距离

地址：http://hihocoder.com/problemset/problem/1478

题目：

水陆距离

时间限制:10000ms

单点时限:1000ms

内存限制:256MB

描述

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

样例输入

4 4  
0110  
1111  
1111  
0110

样例输出

0 1 1 0  
1 2 2 1  
1 2 2 1  
0 1 1 0
思路：bfs！把所有水域点先加入队列，然后按层bfs！

#include <bits/stdc++.h>

using namespace std;

#define MP make_pair
#define PB push_back
typedef long long LL;
typedef pair<int,int> PII;
const double eps=1e-8;
const double pi=acos(-1.0);
const int K=1e6+7;
const int mod=1e9+7;

int n,m,vis[808][808],dis[808][808];
char mp[808][808];
queue<PII >q;
int dx[]={0,0,-1,1},dy[]={-1,1,0,0};
bool check(int x,int y)
{
    return x>=1&&x<=n&&y>=1&&y<=m;
}
void bfs(void)
{
    while(q.size())
    {
        for(int i=1,sz=q.size();i<=sz;i++)
        {
            int x=q.front().first,y=q.front().second;
            q.pop(),vis[x][y]=0;
            for(int i=0;i<4;i++)
            {
                int nx=x+dx[i],ny=y+dy[i];
                if(check(nx,ny)&&!vis[nx][ny]&&dis[nx][ny]>dis[x][y]+1)
                    q.push(MP(nx,ny)),vis[nx][ny]=1,dis[nx][ny]=dis[x][y]+1;
            }
        }
    }
}
int main(void)
{
    memset(dis,0x3f3f3f3f,sizeof dis);
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        scanf("%s",&mp[i][1]);
        for(int j=1;j<=m;j++)
        if(mp[i][j]=='0')
            dis[i][j]=0,q.push(MP(i,j));
    }
    bfs();
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        printf("%d ",dis[i][j]);
        puts("");
    }
    return 0;
}