Leetcode542：矩阵（广度遍历解法）
原创Leetcode542：矩阵（广度遍历解法）
原创
伯约同学
发布于 2022-03-09 22:08:17
发布于 2022-03-09 22:08:17
给定一个由 0 和 1 组成的矩阵 mat ，请输出一个大小相同的矩阵，其中每一个格子是 mat 中对应位置元素到最近的 0 的距离。
两个相邻元素间的距离为 1 。
答题
/**
?
* @param {number[][]} mat
?
* @return {number[][]}
*/
var updateMatrix = function(mat) {
let map = new Array(mat.length); // 用于输出的数组
let quque = []; // 队列
/* 给map赋初值*/
for(let i = 0; i < mat.length; i++){
map[i] = new Array(mat[0].length).fill(Infinity);
for(let j = 0; j < mat[0].length; j++){
if(mat[i][j] == 0){
quque.push([i,j]);
map[i][j] = 0;
}
}
}
/*遍历0周边的数，如果大于0+1，则赋值，将此推进队列，继续遍历*/
while(quque.length > 0){
let size = quque.length;
for(let k = 0; k < size; k++){
let [i,j] = quque.shift();
?
if(i - 1 >= 0 && map[i - 1][j] > map[i][j] + 1){
map[i - 1][j] = map[i][j] + 1;
quque.push([i - 1,j])
}
if(i + 1 < map.length && map[i + 1][j] > map[i][j] + 1){
map[i + 1][j] = map[i][j] + 1;
quque.push([i + 1,j])
}
if(j - 1 >= 0 && map[i][j - 1] > map[i][j] + 1){
map[i][j - 1] = map[i][j] + 1;
quque.push([i,j - 1])
}
if(j + 1 < map[0].length && map[i][j + 1] > map[i][j] + 1){
map[i][j + 1] = map[i][j] + 1;
quque.push([i,j + 1])
}
}
?
}
return map;
};不像其他的，这里是需要一开始就把所有的0位置放到quque中去当做一个集合，与岛屿大小问题大同小异
原创声明：本文系作者授权腾讯云开发者社区发表，未经许可，不得转载。
如有侵权，请联系 cloudcommunity@tencent.com 删除。
原创声明：本文系作者授权腾讯云开发者社区发表，未经许可，不得转载。
如有侵权，请联系 cloudcommunity@tencent.com 删除。
评论
登录后参与评论
推荐阅读