给定一个由 0 和 1 组成的矩阵 mat ,请输出一个大小相同的矩阵,其中每一个格子是 mat 中对应位置元素到最近的 0 的距离。
两个相邻元素间的距离为 1 。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/01-matrix
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
import java.util.LinkedList; import java.util.Queue; class Solution { private int[] dx = {0, 1, 0, -1}; private int[] dy = {1, 0, -1, 0}; public int[][] updateMatrix(int[][] mat) { if (mat == null || mat.length == 0 || mat[0].length == 0) { return new int[0][0]; } int n = mat.length, m = mat[0].length; int[][] ret = new int[n][m]; Queue<Operation> queue = new LinkedList<>(); for (int i = 0; i < n; ++i) { for (int j = 0; j < m; ++j) { if (mat[i][j] == 0) { queue.offer(new Operation(i, j, 0)); } } } while (!queue.isEmpty()) { Operation operation = queue.poll(); ret[operation.x][operation.y] = operation.step; for (int i = 0; i < 4; ++i) { int x = dx[i] + operation.x; int y = dy[i] + operation.y; if (x >= 0 && x < n && y >= 0 && y < m && mat[x][y] == 1) { queue.offer(new Operation(x, y, operation.step + 1)); mat[x][y] = 0; } } } return ret; } } class Operation { int x; int y; int step; public Operation(int x, int y, int step) { this.x = x; this.y = y; this.step = step; } }