hihoCoder 1478 水陆距离

描述

给定一个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

从水域出发进行搜索，将所有水域坐标放入队列，然后进行spfa算法操作，求出到水域最短距离。

我这里没有标记是否进入队列，实际上也就退化成了BellmanFord算法，但是这里点与相邻点距离都是1，

挪动次数等于走的距离，每个点只进入队列一次，所以时间复杂度并没有影响。。

#include <bits/stdc++.h>
using namespace std;

const int maxn=1e5+7;
typedef long long ll;

char g[1000][1000];
int n,m;
int dir[4][2]={{0,1},{1,0},{0,-1},{-1,0}};
int ans[1000][1000];

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=0;i<n;i++){
        scanf("%s",g[i]);
    }
    for(int i=0;i<n;i++){
        fill(ans[i],ans[i]+m+3,10000000);
    }
    queue<pair<int,int > >que;
    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            if(g[i][j]=='0'){ans[i][j]=0;que.push(make_pair(i,j));}
        }
    }
    while(!que.empty()){
        pair<int,int >  p=que.front();
        que.pop();
        for(int i=0;i<4;i++){
            int xx=p.first+dir[i][0];
            int yy=p.second+dir[i][1];
            if(xx>=0&&xx<n&&yy>=0&&yy<m&&ans[p.first][p.second]+1<ans[xx][yy]){
                ans[xx][yy]=ans[p.first][p.second]+1;
                que.push(make_pair(xx,yy));
            }
        }
    }
    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            printf("%d",ans[i][j]);
            if(j!=m-1)printf(" ");
        }
        printf("\n");
    }
    return 0;
}