LeetCode542. 01 Matrix题解

1. 题目描述

Given a matrix consists of 0 and 1, find the distance of the nearest 0 for each cell. The distance between two adjacent cells is 1.
【翻译过来】：给了一个只包含0和1的矩阵，我们需要计算在原矩阵中每个cell到最近的0的距离是多少，两个cell之间的距离是1，并且计算cell的时候只可以按照上下左右的方式计算。

2. 样例输出

Input1:
    0 0 0
    0 1 0
    0 0 0

Output1:
    0 0 0
    0 1 0
    0 0 0

Input2:
    0 0 0
    0 1 0
    1 1 1

Output2:
    0 0 0
    0 1 0
    1 2 1

3. 分析

我们决定利用BFS的方法搜索整个矩阵。
首先搜索矩阵中0元素的位置坐标，将其记录下来，存储在队列里面与此同时，我们生成一个isVisited矩阵，将其初始化为false，代表矩阵中的每个元素是否被访问过。按照这种方式，我们的队列里面就能存储了所有元素值为0的元素坐标。
接下来我们开始遍历队列，将队列中的每个元素（是一个pair，代表0元素位置的横纵坐标matrix[x][y]）按照上下左右四个方向（matrix[x-1][y]，matrix[x+1][y]，matrix[x][y-1]与matrix[x][y+1]）进行搜索。
如果该四个方向的元素不是0，并且未被访问过，则将其坐标也推入到队列中，与此同时，将矩阵该位置的元素值（matrix[x-1][y]或matrix[x+1][y]或matrix[x][y-1]或matrix[x][y+1]）变为matrix[x][y]+1，代表需要在原有的元素位置向上下左右移动一个才能达到。按照这种方式，我们的队列里面就能存储了所有元素值为1的元素坐标。并且元素坐标对应位置的元素就是最终需要得到的结果。

4. 源码

class Solution {
public:
    vector<vector<int>> updateMatrix(vector<vector<int>>& matrix) {
        int M = matrix.size(), N = matrix[0].size();
        int dirs[4][2] = {{-1, 0}, {1, 0}, {0, -1}, {0, 1}};
        vector<vector<bool> >isVisited(M, vector<bool>(N, false));
        queue<pair<int, int> > q;
        for (int i = 0; i < M;i++) {
            for (int j = 0; j < N;j++) {
                isVisited[i][j] = false;
                if (matrix[i][j] == 0) {
                    q.push(make_pair(i, j));
                }
            }
        }

        while(q.empty() == false) {
            int size = q.size();
            for (int m = 0; m < size; m++) {
                int x = q.front().first;
                int y = q.front().second;
                q.pop();
                for (int n = 0; n < 4; n++) {
                    int xx = x + dirs[n][0];
                    int yy = y + dirs[n][1];
                    if (xx >= 0 && xx < M && yy >= 0 && yy < N && matrix[xx][yy] == 1 && isVisited[xx][yy] == false) {
                        isVisited[xx][yy] = true;
                        q.push(make_pair(xx, yy));
                        matrix[xx][yy] = matrix[x][y] + 1;
                    }
                }
            }
        }
        return matrix;
    }
};

5. 心得

BFS的算法，祝愿机考顺利~