一 问题描述
01 矩阵
给定一个由 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标准模板进行求解。
三 问题求解
class Solution {
public int[][] updateMatrix(int[][] matrix) {
int[][] directions={
{1,0},{-1,0},{0,1},{0,-1}};
for(int i=0;i<matrix.length;i++){
mark1:
for(int j=0;j<matrix[0].length;j++){
if(matrix[i][j]==1){
Queue<int[]> queue=new LinkedList<>();