水陆距离 HihoCoder - 1478（bfs+思维）

水陆距离                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表示水域。对于每一个位置，求出它距离最近的水域的距离是多少

思路：原本的思路是将所有水洼的位置存起来，然后分别对每个水洼当起始位置进行bfs，然后比较每个点跟之前的相比的距离，变小则更新。但是这个方法TLE，因为点太多，最多有800*800个，对于每个水洼都要进行bfs，而每次bfs都要进行全图扫描更新，时间复杂度太高，超时也是应该的，这也暴露了我一个毛病，做题不喜欢计算复杂度，要改正。

            正确的思路是：先将水洼的位置入队，然后进行bfs，由于bfs的特性，距离水洼为1的点会先入队，进而是2、3、4...，拿题目的测试案例来说吧，4个0的位置入队后，后续入队的必然是距离这4个0距离为1的点，也就是说距离=1是这些已经入队的点到最近水洼的距离，同理，当vis=1时，说明此点已经被访问过，说明这块陆地已经有距离他最近的水洼，因此只需要一次bfs，无需再更新操作。而我上面的错误想法虽然也只对水洼开始进行bfs，但是每次都需要遍历全图，会浪费极大的时间。

AC代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<utility>
#include<set>
#include<vector>
#include<map>
#include<queue>
#include<stack>
#define maxn 805
#define INF 0x3f3f3f3f
#define LL long long
#define ULL unsigned long long
#define mod 10001
#define P pair<int,int>
using namespace std;

int n,m;
int vis[maxn][maxn];
char maze[maxn][maxn];
int step_num[maxn][maxn];
int nd[4][2] = {{1,0},{-1,0},{0,1},{0,-1}};

struct node{
    int x,y,step;
}nows,nexts;

bool judge(int x,int y)
{
    if(x<1||x>m||y<1||y>n) return false;
    if(vis[x][y]==1)       return false;
    if(step_num[x][y]==0)     return false; //地图中为水塘
    return true;
}

queue<node> q;
void bfs()
{
    memset(vis,0,sizeof(vis));
    while(!q.empty()){
        nows = q.front();
        q.pop();
        for(int i=0;i<4;++i){
            nexts.x = nows.x+nd[i][0];
            nexts.y = nows.y+nd[i][1];
            if(judge(nexts.x,nexts.y)){
                vis[nexts.x][nexts.y] = 1;
                nexts.step = nows.step+1;
                step_num[nexts.x][nexts.y] = nexts.step;
                q.push(nexts);
            }
        }
    }
}
int main()
{
    while(scanf("%d %d",&n,&m)!=EOF){
        int k=0;
        for(int i=1;i<=n;++i){
            for(int j=1;j<=m;++j){
                cin>>maze[i][j];
                step_num[i][j] = -1;  
                if(maze[i][j]=='0'){
                    nows.x = i;
                    nows.y = j;
                    step_num[i][j] = 0;
                    q.push(nows);
                }
            }
        }
        bfs();
        for(int i=1;i<=n;++i){
            for(int j=1;j<=m;++j){
                printf("%d ",step_num[i][j]);
            }
            printf("\n");
        }
    }
    return 0;
}