01矩阵

题目描述

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