喵哈哈村的魔法考试 Round #7 (Div.2) B

	B	喵哈哈村的麦克雷

喵哈哈村的麦克雷

发布时间: 2017年3月13日 11:51   最后更新: 2017年3月14日 18:16   时间限制: 1000ms   内存限制: 128M

描述

为了拯救喵哈哈村，这个世界必须要存在英雄。

一名叫做麦克雷的英雄站了出来！他现在面临一个难题：

给定一个N x M的01矩阵，其中1表示陆地，0表示水域。对于每一个位置，求出它距离最近的水域的距离是多少。  

矩阵中每个位置与它上下左右相邻的格子距离为1。

输入

本题包含若干组测试数据：
第一行包含两个整数，N和M。
以下N行每行M个0或者1，代表地图。
数据保证至少有1块水域。
满足，1 <= N, M <= 100

输出

输出N行，每行M个空格分隔的整数。每个整数表示该位置距离最近的水域的距离。
每行的末尾都请加一个空格……

样例输入1 复制

4 4  
0110  
1111  
1111  
0110

样例输出1

0 1 1 0  
1 2 2 1  
1 2 2 1  
0 1 1 0

Hint: bfs
一开始看到这个题目，就想到bfs搜距离嘛...然后就从第一个开始去搜 搜 搜...搜到最后一个点，样例是通过了 然后发现tle了...
看到题解之后。。。噢原来可以这样做啊...（把0放进队列，然后从0开始搜嘛，每次从同一层次级别开始搜） bfs的特点是层序遍历，所以它搜到的就是最短路（可以用反证法证明一下）

#include<iostream>
#include<queue>
#include<cstdio>
#include<cstring>
using namespace std;
typedef pair<int, int> p;
char g[1000][1000];
int sign[1000][1000];
int dx[]={1, 0, -1, 0}, dy[]={0, 1, 0, -1};
int n, m;
inline bool ok(int x, int y){
    if(x>=0&&x<n&&y>=0&&y<m) return true;
    return false;
}
/*
inline void bfs(int x, int y){
    
    que.push(make_pair(x, y));
    while(que.size()){
        p now=que.front(); que.pop();
        for(int i=0; i<4; i++){
            int nx=now.first+dx[i], ny=now.second+dy[i];
            if(ok(nx, ny))    {
                que.push(make_pair(nx, ny));
                step[nx][ny]=step[now.first][now.second]+1;
                if(g[nx][ny]=='0'){
                    sign[x][y]=step[nx][ny];
                    sign[now.first][now.second]=1;
                    return;
                }    
            }    
        }
    }
}
*/
int main(){
    while(cin>>n>>m){
        for(int i=0; i<n; i++){
                scanf("%s", &g[i]);
        }
        memset(sign, -1, sizeof(sign));
        queue<pair<int, int> >que;
        for(int i=0; i<n; i++){
            for(int j=0; j<m; j++){
                if(g[i][j]=='0'){
                    sign[i][j]=0;
                    que.push(make_pair(i, j));
                }
            }
        }
        while(que.size()){
            p now=que.front(); que.pop();
            for(int i=0; i<4; i++){
                int nx=now.first+dx[i], ny=now.second+dy[i];
                if(!ok(nx, ny))    continue;
                if(sign[nx][ny]!=-1)    continue;
                sign[nx][ny]=sign[now.first][now.second]+1;
                que.push(make_pair(nx, ny));
            }
        }
        for(int i=0; i<n; i++){
            for(int j=0; j<m; j++){
                printf("%d ", sign[i][j]);
            }
            printf("
");
        }
    }
    
    return 0;
}