题目描述
给定一个由 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 1 1
注意:
给定矩阵的元素个数不超过 10000。
给定矩阵中至少有一个元素是 0。
矩阵中的元素只在四个方向上相邻: 上、下、左、右。
解题思路
1、DP动态规划法
package com.company;
/**
* DP(动态规划)
* 对于每个点,值为0距离为0,
* 值不为0,则值为上下左右点最小的那个加1
*/
public class updateMatrix {
public int[][] updateMatrix(int[][] matrix) {
int row = matrix.length;
int col = matrix[0].length;
// 第一遍遍历,从左上到右下,比较上面和左边中较小的值
// matrix[i][j]代表第(i,j)个点距离0最短的距离
// 毫无疑问,所有的0的位置一定都是0
// 所有1的位置初始化为一个大于1的数(便于后面比较)
// dp[i][j] = min(dp[i][j],min(dp[i-1][j],dp[i][j-1])+1)
for(int i = 0;i < row;i++){
for(int j = 0;j < col;j++){
if(matrix[i][j] == 1){
matrix[i][j] = row + col;
}
if(i > 0){
matrix[i][j] = Math.min(matrix[i][j],matrix[i-1][j] + 1);
}
if(j > 0){
matrix[i][j] = Math.min(matrix[i][j], matrix[i][j - 1] + 1);
}
}
}
// 第二次遍历,反向遍历,根据相邻右元素和下元素及当前元素的结果得出最终结果
for (int i = row - 1; i >= 0; i--) {
for (int j = col - 1; j >= 0; j--) {
if (i < row - 1) {
matrix[i][j] = Math.min(matrix[i][j], matrix[i + 1][j] + 1);
}
if (j < col - 1) {
matrix[i][j] = Math.min(matrix[i][j], matrix[i][j + 1] + 1);
}
}
}
return matrix;
}
}
2、BFS深度优先搜索
待续