题目2 : 水陆距离 时间限制:10000ms 单点时限:1000ms 内存限制:256MB
描述
给定一个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题目一眼扫下去,就知道是使用节点展开的宽度优先搜索算法(bfs),笔者不是太熟练,就使用了两个队列分别push和pop来实现,其实一个队列就可以了,只需要记下来队列原来的size即可。
#include <iostream> #include <queue> using namespace std; int x[4]={0,1,0,-1}; int y[4]={-1,0,1,0}; int flag=0; int main() { int n,m; cin>>n>>m; int b[1000][1000]; getchar(); char a; queue <pair <int,int>> qu; queue <pair <int,int>> qu2; for(int i=1;i<=n;i++) { for(int j=1;j<=m;j++) { scanf("%c",&a); if(a=='0') { b[i][j]=0; qu.push(make_pair(i,j)); } else b[i][j]=m+n+10086; } getchar(); } while(true) { flag++; while(!qu.empty()) { int xx=qu.front().first; int yy=qu.front().second; for(int i=0;i<=3;i++) { if(b[xx+x[i]][yy+y[i]]==m+n+10086&&(xx+x[i])>=1&&(yy+y[i])>=1&&(xx+x[i])<=n&&(yy+y[i])<=m) { b[xx+x[i]][yy+y[i]]=flag; qu2.push(make_pair(xx+x[i],yy+y[i])); } } qu.pop(); } flag++; while(!qu2.empty()) { int xx=qu2.front().first; int yy=qu2.front().second; for(int i=0;i<=3;i++) { if(b[xx+x[i]][yy+y[i]]==m+n+10086&&(xx+x[i])>=1&&(yy+y[i])>=1&&(xx+x[i])<=n&&(yy+y[i])<=m) { b[xx+x[i]][yy+y[i]]=flag; qu.push(make_pair(xx+x[i], yy+y[i])); } } qu2.pop(); } if(qu.empty()&&qu2.empty()) break; } for(int i=1;i<=n;i++) { for(int j=1;j<m;j++) { cout<<b[i][j]<<" "; } cout<<b[i][m]<<endl; } }wa了一发,因为输入的图里面居然没有空格,这个真的有的惊讶,233333,太粗心了。