542:01 矩阵

src/main/java/leetcode/editor/cn/Zero1Matrix.java

@@ -0,0 +1,101 @@
+//给定一个由 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)
+
+}

src/main/java/leetcode/editor/cn/Zero1Matrix.md

@@ -0,0 +1,44 @@
+<p>给定一个由 0 和 1 组成的矩阵，找出每个元素到最近的 0 的距离。</p>
+
+<p>两个相邻元素间的距离为 1 。</p>
+
+<p> </p>
+
+<p><b>示例 1：</b></p>
+
+<pre>
+<strong>输入：</strong>
+[[0,0,0],
+ [0,1,0],
+ [0,0,0]]
+
+<strong>输出：</strong>
+[[0,0,0],
+ [0,1,0],
+ [0,0,0]]
+</pre>
+
+<p><b>示例 2：</b></p>
+
+<pre>
+<b>输入：</b>
+[[0,0,0],
+ [0,1,0],
+ [1,1,1]]
+
+<strong>输出：</strong>
+[[0,0,0],
+ [0,1,0],
+ [1,2,1]]
+</pre>
+
+<p> </p>
+
+<p><strong>提示：</strong></p>
+
+<ul>
+	<li>给定矩阵的元素个数不超过 10000。</li>
+	<li>给定矩阵中至少有一个元素是 0。</li>
+	<li>矩阵中的元素只在四个方向上相邻: 上、下、左、右。</li>
+</ul>
+<div><div>Related Topics</div><div><li>广度优先搜索</li><li>数组</li><li>动态规划</li><li>矩阵</li></div></div>\n<div><li>👍 446</li><li>👎 0</li></div>