LeetCode-01矩阵（宽度优先）

（一）题目描述

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

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

（二）解题思路

以每个0元素为中心向四周扩散，涉及到的点均更新其到0元素的最短距离，即先将所有值为0的元素的加入队列，然后依次取出并广度遍历该取出元素的四个方向，更新四个方向的最短距离，只要初始值为INT_MAX的点均会被更新并加入队列。

（三）代码如下：

class Solution07 {
    public int[][] updateMatrix(int[][] mat) {
        int INT_MAX=1000000;
        int rowLen=mat.length;
        int colLen=mat[0].length;
        int[][] dist=new int[rowLen][colLen];
        Queue<int[]> queue = new LinkedList<int[]>();


//        (1)首先从矩阵中的0开始向外搜索，因为与0相邻的四个方向的非0元素位置距离为1;从0开始搜索就能得到更多的信息
        for (int i = 0; i < rowLen; i++) {
            for (int j = 0; j < colLen; j++) {
                dist[i][j]=INT_MAX;
                if(mat[i][j]==0){
                    //将所有值为0的位置入队列
                    queue.offer(new int[]{i,j});
//                    所有值为0的位置其距离也置为0
                    dist[i][j]=0;
                }
            }
        }

//        距离矩阵中的每个元素均会入队列，因为每个元素均会被搜索四个方向
        int[][] directions={{-1,0},{0,1},{1,0},{0,-1}};
        while (!queue.isEmpty()){
//            队头元素出队
            int[] arr=queue.poll();
            //取出元素的横纵坐标
            int m=arr[0];
            int n=arr[1];
            for (int i = 0; i < 4; i++) {
                //得到新的坐标
                int newi=m+directions[i][0];
                int newj=n+directions[i][1];
                //判断横纵坐标是否越界
                if(newi>=0&&newi<rowLen&&newj>=0&&newj<colLen){
                    //核心！遍历当前取出的元素的四个方向，如果当前取出元素+1（即为相邻元素）要比原来在该位置元素的值要小，则说明当前找的路径更短，则替换
                    if(dist[newi][newj]>dist[m][n]+1){
                        dist[newi][newj]=dist[m][n]+1;
                        queue.offer(new int[]{newi,newj});
                    }
                }
            }
        }

        return dist;
    }
}