给定一个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
思路:搜索,但是不能直接从陆地到水域的搜,这样搜的话每个陆地都要进行一遍搜索,会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");
}
}
}