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

思路：搜索，但是不能直接从陆地到水域的搜，这样搜的话每个陆地都要进行一遍搜索，会TLE。正确的想法是从每个水域同时进行搜索，因为每走一步都会标记，所以不会出现同一个陆地搜索多次的情况；

下面代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#include<stack>
#include<map>
#include<vector>
#define ll long long
using namespace std;
const int MAXN=160000+10;
const int MAXM=800+10;
const int inf=0x3f3f3f;
int xx[MAXN];
int mp[MAXM][MAXM];
int n,m;
char s[MAXM][MAXM];
int vist[4][2]= {{1,0},{-1,0},{0,1},{0,-1}};
int vast[MAXM][MAXM];
struct node
{
    int x,y;
    int ans;
};
node xy,xz;
queue<node>q;
int judge(int x,int y)
{
    if(x>=0&&x<n&&y>=0&&y<m&&!vast[x][y])
        return 1;
    return 0;
}
void bfs()
{
    while(!q.empty())
    {
        xy=q.front();
        q.pop();
        for(int i=0; i<4; i++)
        {
            int xi=xy.x+vist[i][0];
            int yi=xy.y+vist[i][1];
            if(judge(xi,yi))
            {
                xz.x=xi;
                xz.y=yi;
                xz.ans=xy.ans+1;
                mp[xi][yi]=xz.ans;
                q.push(xz);
                vast[xi][yi]=1;
            }
        }
    }
}
int main()
{
    while(cin>>n>>m)
    {
        while(!q.empty()) q.pop();
        memset(vast,0,sizeof(vast));
        for(int i=0; i<n; i++)
        {
            scanf("%s",s[i]);
            for(int j=0; j<m; j++)
            {
                if(s[i][j]=='0')/将水域进入队列，以便后续搜索
                {
                    vast[i][j]=1;
                    xy.x=i;
                    xy.y=j;
                    xy.ans=0;
                    q.push(xy);
                }
            }
        }
        memset(mp,0,sizeof(mp));
        bfs();
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<m;j++)
            {
                if(j==0) printf("%d",mp[i][j]);
                else printf(" %d",mp[i][j]);
            }
            printf("\n");
        }
    }
}