LeetCode 刷题 （七）算法入门

542. 01 矩阵

给定一个由 0 和 1 组成的矩阵 mat ，请输出一个大小相同的矩阵，其中每一个格子是 mat 中对应位置元素到最近的 0 的距离。

两个相邻元素间的距离为 1 。

方法一：思路：首先，我们很容易理解，从每一个0开始去计算到不为0的dist，采用广度优先的遍历，第一轮被遍历到的就是1，从每一个0的位置开始依次BFS进行，然后取dist的最小值。不过如果0的个数较多，处理起来比较复杂。在思考一下，我们可以提升一下思维，我们可以假设一个BFS的起点（逻辑上存在），这个起点和矩阵中所有的0相连。这样，我们只需要对这个逻辑上的起点进行一次BFS，BFS的第一轮，我们将所有的0的位置入队，然后接下来第二轮就是距离0的距离为1的元素。

class Solution {
private:
    static constexpr int dirs[4][2] = {{-1, 0}, {1, 0}, {0, -1}, {0, 1}};

public:
    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);
                    seen[i][j] = 1;
                }
            }
        }

        // 广度优先搜索
        while (!q.empty()) {
            auto [i, j] = q.front();
            q.pop();
            for (int d = 0; d < 4; ++d) {
                int ni = i + dirs[d][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;
                    q.emplace(ni, nj);
                    seen[ni][nj] = 1;
                }
            }
        }

        return dist;
    }
};

        还有后续。。。。。。。