[路飞]_js算法：leetcode 542-01 矩阵

leetcode 542. 01 矩阵

问题描述： 给定一个由 0 和 1 组成的矩阵 mat ，请输出一个大小相同的矩阵，其中每一个格子是 mat 中对应位置元素到最近的 0 的距离。

两个相邻元素间的距离为 1 。

示例 1：

输入： mat = [[0,0,0],[0,1,0],[0,0,0]]
输出： [[0,0,0],[0,1,0],[0,0,0]]

示例 2：

输入： mat = [[0,0,0],[0,1,0],[1,1,1]]
输出： [[0,0,0],[0,1,0],[1,2,1]]

提示：

• m == mat.length
• n == mat[i].length
• 1 <= m, n <= 104
• 1 <= m * n <= 104
• mat[i][j] is either 0 or 1.
• mat 中至少有一个 0 

思路： 广度优先遍历

//数据状态设计-队列节点数据
function Data(i,j,k){
  this.i=i||0;//行
  this.j=j||0;//列
  this.k=k||0;//当前节点距离最近的0的距离
}
function initQueue(queue,vis,m,n,mat){
  for(let i=0;i<m;i++){
    vis.push(new Array(n).fill(-1))
  }
  for(let i=0;i<m;i++){
    for(let j=0;j<n;j++){
      if(mat[i][j])continue;
      vis[i][j]=0;
      queue.push(new Data(i,j,0))//把所有是0的节点添加到队列中
    }
  }
}
 var updateMatrix = function(mat) {
   let m=mat.length;//行
   let n=mat[0].length;//列
   let vis=[];//设置一个同样的矩阵，来判断节点是否遍历过，为了及时止损。
   let queue=[];
   initQueue(queue,vis,m,n,mat);
   let vec=[[-1,0],[1,0],[0,-1],[0,1]];//方向数组
   while(queue.length){
     let cur=queue[0];
    for(let i=0;i<vec.length;i++){
      let x=cur.i+vec[i][0];
      let y=cur.j+vec[i][1];
      if(x<0||x>=m)continue;//边界判断
      if(y<0||y>=n)continue;//边界判断
      if(vis[x][y]!=-1)continue;//边界判断（访问过）
      vis[x][y]=cur.k+1;
      queue.push(new Data(x,y,cur.k+1))
    }
    queue.shift()
   }
   return vis;
};