题干
给定一个由 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。
矩阵中的元素只在四个方向上相邻: 上、下、左、右。
想法
首先确定题干说的距离是路径的距离,不是直线距离
这道题首先想到bfs
这篇写得很好,看它就行传送门
我是用的动态规划
首先初始化一个dp[][]
原本矩阵为0的位置肯定对应dp也为0
对于1的位置
每一步只能上下左右选一次
如果往左上走到了
那么最短的是
dp[i][j] = Math.min(dp[i][j-1], dp[i - 1][j]) + 1;
右下同理
只要这两个方向就行了
看代码
package daily;
import Exam.Up;
import java.util.Arrays;
public class UpdateMatrix {
public int[][] updateMatrix(int[][] matrix) {
//0处理
if (matrix == null || matrix.length == 0) {
return null;
}
int m = matrix.length;
int n = matrix[0].length;
int[][] dp = new int[m][n];
//初始化 0为0 其余的为最大值
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (matrix[i][j] == 0) {
dp[i][j] = 0;
} else {
dp[i][j] = 10001;
}
}
}
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (i - 1 >= 0) {
//上
dp[i][j] = Math.min(dp[i][j], dp[i - 1][j] + 1);
}
if (j - 1 >= 0) {
//左
dp[i][j] = Math.min(dp[i][j], dp[i][j - 1] + 1);
}
}
}
for (int i = m - 1; i >= 0; i--) {
for (int j = n - 1; j >= 0; j--) {
if (i <= m - 2) {
//下
dp[i][j] = Math.min(dp[i][j], dp[i + 1][j] + 1);
}
if (j <= n - 2) {
//右
dp[i][j] = Math.min(dp[i][j], dp[i][j + 1] + 1);
}
}
}
return dp;
}
public static void main(String[] args) {
UpdateMatrix updateMatrix = new UpdateMatrix();
int[][] test1 = {{0, 0, 0}, {0, 1, 0}, {1, 1, 1}};
int[][] res = updateMatrix.updateMatrix(test1);
for (int[] tem:res
) {
System.out.println(Arrays.toString(tem));
}
}
}