给定一个由0和1组成的矩阵,找出每个元素到最近的 0 的距离。
两个相邻元素间的距离为 1 。
题解:
1.01矩阵
2.距离最近0的距离
3.矩阵中至少有一个0元素
示例 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。
矩阵中的元素只在四个方向上相邻: 上、下、左、右。
解题思路:
附设两个辅助vector和一个队列,实现广度优先遍历
-
两个数组分别用来记录是否访问过、题目要求的距离
-
队列存放矩阵中0元素的下标
-
遍历原矩阵中的每一个元素,如果该元素的四个方向上有未访问过的(或在原矩阵位置上为1),该方向位置上的存储距离的矩阵元素值比遍历元素距离矩阵元素值多1个单位,并记录方向元素下标,置该元素访问过
-
访问数组遍历完,距离矩阵中对应位置距离也计算完毕
C/C++题解:
class Solution {
public:
int dirs[4][2] = {{-1, 0}, {1, 0}, {0, -1}, {0, 1}};//四个方向,可以扩展到8方向
vector<vector<int>> updateMatrix(vector<vector<int>>& matrix) {
int m = matrix.size(), n = matrix[0].size();
vector<vector<int>> dist(m, vector<int>(n));
vector<vector<int>> seen(m, vector<int>(n));
queue<pair<int, int>> q;
// 将所有的 0 添加进初始队列中
for (int i = 0; i < m; ++i) {
for (int j = 0; j < n; ++j) {
if (matrix[i][j] == 0) {
q.emplace(i, j);//记录值为0的下标
seen[i][j] = 1;
}}}//seen中值为0的位置为matrix中值为1的
// 广度优先搜索
while (!q.empty()) {//取队列中记录的值为0的元素下标
auto [i, j] = q.front();
q.pop();//并从队列中删除
for (int d = 0; d < 4; ++d) {//从头遍历元素(i,j)四个方向的值
int ni = i + dirs[d][0];//如果发现seen中有一个方向上的元素为0
int nj = j + dirs[d][1];
if (ni >= 0 && ni < m && nj >= 0 && nj < n && !seen[ni][nj]) {
dist[ni][nj] = dist[i][j] + 1;//该方向位置比(i,j)多一个距离
q.emplace(ni, nj);//再添加到seen一起的队列中
seen[ni][nj] = 1;}}}//用该方向位置上的距离赋值
return dist;}};//遍历完seen数组,dist中即是计算完的距离
Debug结果:
更多题解可前往公众号免费获取