给定一个由 0 和 1 组成的矩阵，找出每个元素到最近的 0 的距离。
两个相邻元素间的距离为 1 。
示例 1:
输入:
0 0 0
0 1 0
0 0 0
输出:
0 0 0
0 1 0
0 0 0
bfs代码
class Solution {int[][] res;public int[][] updateMatrix(int[][] matrix) {int[][] dir=new int[][]{{-1,0},{1,0},{0,1},{0,-1}};res=new int[matrix.length][matrix[0].length];LinkedList<int[]> queue=new LinkedList<>();for(int i=0;i<matrix.length;i++)for(int j=0;j<matrix[0].length;j++)if(matrix[i][j]==0)//将0元素入队{res[i][j]=0;matrix[i][j]=-1;queue.add(new int[]{i,j});}while (!queue.isEmpty()){int[] p=queue.removeFirst();int x=p[0],y=p[1];for(int[] d:dir)//4个方向入队{int nextX=x+d[0],nextY=y+d[1];if(nextX<0||nextY<0||nextX>=matrix.length||nextY>=matrix[0].length||matrix[nextX][nextY]==-1)//不符合的情况剔除continue;res[nextX][nextY]=res[x][y]+1;matrix[nextX][nextY]=-1;queue.add(new int[]{nextX,nextY});}}return res;}
}
动态规划代码
class Solution {public int[][] updateMatrix(int[][] matrix) {int n=matrix.length,m=matrix[0].length;int[][] dp=new int[n][m];for(int i=0;i<n;i++)for(int j=0;j<m;j++)dp[i][j]=matrix[i][j]==0?0:100001;for(int i=0;i<n;i++)for(int j=0;j<m;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=n-1;i>=0;i--)for(int j=m-1;j>=0;j--)//从右和下推导{ if(i+1<n)dp[i][j]= Math.min(dp[i][j],dp[i+1][j]+1);if(j+1<m)dp[i][j]= Math.min(dp[i][j],dp[i][j+1]+1);} return dp;}
}
)
