题目描述
给定一个由 0 和 1 组成的矩阵,找出每个元素到最近的 0 的距离。
两个相邻元素间的距离为 1 。
输入
第一行两个整数n m ( n <= 100 m <= 100 )
接下来输入 n行 m列 的二维数组。
3 3
0 0 0
0 1 0
0 0 0
输出
输出每个元素到最近的 0 的距离。
0 0 0
0 1 0
0 0 0
示例
输入:
3 3
0 0 0
0 1 0
1 1 1
输出:
0 0 0
0 1 0
1 2 1
思路
定义一个队列,初始时将所有 0 元素的坐标进队;0 元素位置的值为 0,其余为 -1。(也可以定义为其它数字,只是为了保证每个1只被便利一次)
对这个队列开始进行宽度优先搜索,每次扩展上下左右四个方向,若发现新的位置为 -1,则更新新位置的答案为当前位置答案加 1。
代码
#include<bits/stdc++.h>
using namespace std;
int n,m;
int a[100][100],b[100][100],v[4][2]={1,0,-1,0,0,1,0,-1};;
struct node
{
int x;
int y;
int step;
node(int xx,int yy,int step1):x(xx),y(yy),step(step1){}
};
queue<node>q;
void bfs()
{
while(!q.empty())
{
node s=q.front();
for(int i=0;i<4;i++)
{
int tx=s.x+v[i][0];
int ty=s.y+v[i][1];
if(tx>=0&&tx<n&&ty>=0&&ty<m&&a[tx][ty]==-1)
{
q.push(node(tx,ty,s.step+1));
a[tx][ty]=s.step+1;
}
}
q.pop();
}
}
int main()
{
cin>>n>>m;
for(int i=0;i<n;i++)
for(int j=0;j<m;j++)//输入矩阵
{
cin>>a[i][j];
if(a[i][j]==0)
q.push(node(i,j,0));
if(a[i][j]==1)
a[i][j]=-1;//这里必须用一个不同于1的数标记,为了避免1被多次扫到
}
bfs();
for(int i=0;i<n;i++)
{
for(int j=0;j<m;j++)
cout<<a[i][j]<<" ";
cout<<endl;
}
return 0;
}