101 lines
2.5 KiB
Java
101 lines
2.5 KiB
Java
|
//给定一个由 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。
|
|||
|
|
// 矩阵中的元素只在四个方向上相邻: 上、下、左、右。
|
|||
|
|
//
|
|||
|
|
// Related Topics 广度优先搜索 数组 动态规划 矩阵
|
|||
|
|
// 👍 446 👎 0
|
|||
|
|
|
|||
|
|
package leetcode.editor.cn;
|
|||
|
|
|
|||
|
|
import javafx.util.Pair;
|
|||
|
|
|
|||
|
|
import java.util.Arrays;
|
|||
|
|
import java.util.LinkedList;
|
|||
|
|
import java.util.List;
|
|||
|
|
import java.util.Queue;
|
|||
|
|
|
|||
|
|
//542:01 矩阵
|
|||
|
|
public class Zero1Matrix {
|
|||
|
|
public static void main(String[] args) {
|
|||
|
|
//测试代码
|
|||
|
|
Solution solution = new Zero1Matrix().new Solution();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//力扣代码
|
|||
|
|
//leetcode submit region begin(Prohibit modification and deletion)
|
|||
|
|
class Solution {
|
|||
|
|
public int[][] updateMatrix(int[][] mat) {
|
|||
|
|
List<Pair<Integer, Integer>> list = Arrays.asList(
|
|||
|
|
new Pair<>(-1, 0),
|
|||
|
|
new Pair<>(1, 0),
|
|||
|
|
new Pair<>(0, -1),
|
|||
|
|
new Pair<>(0, 1)
|
|||
|
|
);
|
|||
|
|
Queue<int[]> queue = new LinkedList<>();
|
|||
|
|
boolean[][] is = new boolean[mat.length][mat[0].length];
|
|||
|
|
for (int i = 0; i < mat.length; i++) {
|
|||
|
|
for (int j = 0; j < mat[0].length; j++) {
|
|||
|
|
if (mat[i][j] == 0) {
|
|||
|
|
queue.offer(new int[]{i, j});
|
|||
|
|
is[i][j] = true;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
while (!queue.isEmpty()) {
|
|||
|
|
int[] arr = queue.poll();
|
|||
|
|
for (Pair<Integer, Integer> pair : list) {
|
|||
|
|
int x = arr[0] + pair.getKey();
|
|||
|
|
int y = arr[1] + pair.getValue();
|
|||
|
|
if (x < 0 || x >= mat.length || y < 0 || y >= mat[0].length || is[x][y] || mat[x][y] != 1) {
|
|||
|
|
continue;
|
|||
|
|
}
|
|||
|
|
mat[x][y] = mat[arr[0]][arr[1]] + 1;
|
|||
|
|
is[x][y] = true;
|
|||
|
|
queue.offer(new int[]{x, y});
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return mat;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
//leetcode submit region end(Prohibit modification and deletion)
|
|||
|
|
|
|||
|
|
}
|