22.0-1矩阵
给定一个由 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。
矩阵中的元素只在四个方向上相邻: 上、下、左、右。
class Solution {
public:
vector<vector<int>> updateMatrix(vector<vector<int>>& matrix) {
int r,c;
r=matrix.size();
c=matrix[0].size();
vector<vector<int>> f(r,vector<int>(c,10005));
for(int i=0;i<r;i++)
{
for(int j=0;j<c;j++)
{
if(matrix[i][j]==0){f[i][j]=0;continue;}
if(j>=1) f[i][j]=min(f[i][j],f[i][j-1]+1);
if(i>=1) f[i][j]=min(f[i][j],f[i-1][j]+1);
}
}
for(int i=r-1;i>=0;i--)
{
for(int j=c-1;j>=0;j--)
{
if(j+1<c) f[i][j]=min(f[i][j],f[i][j+1]+1);
if(i+1<r) f[i][j]=min(f[i][j],f[i+1][j]+1);
}
}
return f;
}
};