LeetCode542. 01 矩阵

题目

给定一个由 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、把所有的0入队，并先把对应为1的元素初始化为-1，这样可以用matrix同时记录距离和标志是否访问
2、从各个0一层一层的向1不断扩大，直到覆盖整个矩阵

代码

class Solution {
    public int[][] updateMatrix(int[][] matrix) {

         // 四个方向
        int[][] directions = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}}; 
          		  		
        int rows = matrix.length;
		int cols = matrix[0].length;
 
       Queue<int[]> queue = new LinkedList<>();
        for (int i = 0; i < rows; i++) {
            for (int j = 0; j < cols; j++) {
                //1、把所有的0入队
                if (matrix[i][j] == 0) {
                    queue.offer(new int[] {i, j});
                }else{
                    matrix[i][j] = -1;
                }
            }
        }
     
        while (!queue.isEmpty()) {
            int currentQueueSize = queue.size();
              //2、从各个0一层一层的向1不断扩大
            for (int i = 0; i < currentQueueSize; i++) {
                int[] current = queue.poll();
                int tx = current[0];
                int ty = current[1];
                for (int[] direction : directions) {
                    int m = tx + direction[0];
                    int n = ty + direction[1];
                    if (isInGrid(m, n, rows,cols) && matrix[m][n] == -1) {
                        matrix[m][n] = matrix[tx][ty] + 1;
                        queue.offer(new int[] {m, n});
                    }
                }
            }
        }
        return matrix;
    }
  
     //是否在网格区域范围内 
    private boolean isInGrid (int x, int y, int rows,int cols) {
        return 0 <= x && x < rows && 0 <= y && y < cols;
    }
}

复杂度分析

• 时间复杂度：$O(N^2)$，这里 $N$ 是矩阵的边长。二维矩阵里所有的元素都会被搜索一遍；
• 空间复杂度：$O(N^2)$