题目描述:
给定一个由 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。
矩阵中的元素只在四个方向上相邻: 上、下、左、右。
方法1:
主要思路:解题汇总链接
(1)正常的1元素到0元素的距离,和0直接接触的1,距离是1,标记出所有的这些距离1之后,剩余1元素到这些距离1的1元素的距离就是2,标记处所有的2之后,下次剩余的1元素到这些距离为2的1元素的距离就是3,……以此类推,逐渐的增长元素1到0的距离;
(2)使用广度优先模拟这个过程,先存储距离为1的,再处理距离为2的,再处理距离为3的……,为了区分元素,这里先用距离的负数来标识对应的距离,最后再将所有的距离取相反数即可;
class Solution {
public:
vector<vector<int>> updateMatrix(vector<vector<int>>& matrix) {
queue<pair<int,int>> q;
int row=matrix.size();
int col=matrix[0].size();
for(int i=0;i<row;++i){
//初始的将所有的直接和0接触的1元素存储,并标识为-1
for(int j=0;j<col;++j){
if(matrix[i][j]==0