题目描述
给定一个由 0 和 1 组成的矩阵,找出每个元素到最近的 0 的距离。
两个相邻元素间的距离为 1 。
示例 1:
输入:
0 0 0
0 1 0
0 0 0
输出:
0 0 0
0 1 0
0 0 0
示例 2:
输入:
0 0 0
0 1 0
1 1 1
输出:
0 0 0
0 1 0
1 2 1
注意:
给定矩阵的元素个数不超过 10000。
给定矩阵中至少有一个元素是 0。
矩阵中的元素只在四个方向上相邻: 上、下、左、右。
题目分析
题目很容易理解,即是找矩阵中各个元素与0元素的距离。对于0元素自身来说,这个距离自然就是0了,而对于非0元素来说,它对于不同的0元素距离是可能不同的,那么它最终的距离当然是取这些距离中的最小值了。
这样一来,如果某个元素与0元素相邻(指与0元素上下左右相邻),那么该元素与0元素的距离则是1了,如果某个元素不与0元素相邻,那么该元素与0元素的距离就应当取该元素上下左右相邻元素各自与0元素的距离值中的最小值加一,好了,那么这道题该怎么做呢?如果用最笨的方法,即是对每一个元素的上下左右四个方向慢慢扩展,如果碰到0了那么距离也就出来了,不过这种方法显然复杂度是非常高的。
因此就考虑使用BFS来完成,广度优先是使用队列,这里的想法是,先将所有0元素的坐标压入队列中,然后再取出每个0元素的坐标,并且遍历该坐标的上下左右,很显然,如果上下左右中