01 Matrix

题目来源
给一个01矩阵，求矩阵中的1最近的0离它有多远，更新矩阵变为一个距离矩阵。
我想着从左上到右下遍历一遍，从右下到左上遍历一遍。
代码如下：

class Solution {
public:
    vector<vector<int>> updateMatrix(vector<vector<int>>& matrix) {
        int rows = matrix.size(), cols = matrix[0].size();
        vector<vector<int>> dirs{{0, -1}, {-1, 0}, {1, 0}, {0, 1}};
        for (int i=0; i<rows; i++)
            for (int j=0; j<cols; j++) {
                if (matrix[i][j] == 1) {
                    matrix[i][j] = 100000;
                    for (int k=0; k<2; k++) {
                        if (i+dirs[k][0] >= 0 && i+dirs[k][0] < rows && j+dirs[k][1] >= 0 && j+dirs[k][1] < cols)
                            matrix[i][j] = min(matrix[i][j], matrix[i + dirs[k][0]][j + dirs[k][1]] + 1);
                    }
                }
            }
        for (int i=rows-1; i>=0; i--)
            for (int j=cols-1; j>=0; j--) {
                if (matrix[i][j] != 0) {
                    for (int k=2; k<dirs.size(); k++) {
                        if (i+dirs[k][0] >= 0 && i+dirs[k][0] < rows && j+dirs[k][1] >= 0 && j+dirs[k][1] < cols)
                            matrix[i][j] = min(matrix[i][j], matrix[i + dirs[k][0]][j + dirs[k][1]] + 1);
                    }
                }
            }
        return matrix;
    }
};