到最近的0的距离,对每一个非0的位置进行搜索,找到最短的距离即可,但如果对每一个非0的点都进行一次搜索的话,肯定是会超时的。这里可以考虑,将所有0点想象成一个0点(超级0)。然后找到所有1点到超级0的距离即可。
class Solution {
public:
int dx[4] = {0,0,1,-1};
int dy[4] = {1,-1,0,0};
vector<vector<int>> updateMatrix(vector<vector<int>>& mat) {
int n = mat.size(), m = mat[0].size();
vector<vector<int>> dist(n, vector<int>(m, 0));
vector<vector<int>> st(n, vector<int>(m));
queue<pair<int,int>> q;
for (int i = 0; i < n; i++)
{
for (int j = 0; j < m; j++)
{
if (mat[i][j] == 0)
{
st[i][j] = 1;
q.push({i, j});
}
}
}
while (!q.empty())
{
auto t = q.front();
q.pop();
int x = t.first;
int y = t.second;
for (int i = 0; i < 4; i++)
{
int xx = x + dx[i];
int yy = y + dy[i];
if (xx >= 0 && xx < n && yy >= 0 && yy < m && !st[xx][yy])
{
st[xx][yy] = 1;
dist[xx][yy] = dist[x][y] + 1;
q.push({xx, yy});
}
}
}
return dist;
}
};