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;
    }
};
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 这道题不看答案,没想出思路,做题还是不够,看了bfs的思路,然后手写了一遍,记录一下。这道题从bfs的角度来看,思...
    风之羁绊阅读 470评论 0 0
  • Given a matrix consists of 0 and 1, find the distance of ...
    greatseniorsde阅读 293评论 0 0
  • 一前言 特征值 奇异值 二奇异值计算 三PCA 1)数据的向量表示及降维问题 2)向量的表示及基变换 3)基向量 ...
    Arya鑫阅读 10,656评论 2 43
  • Android 自定义View的各种姿势1 Activity的显示之ViewRootImpl详解 Activity...
    passiontim阅读 173,403评论 25 708
  • 目前最困挠我的就是丫头每天早上的坏情绪拉锯战。暑假还在担心初开学又要开始哭闹,上学第一二天的好心情正让我沉浸在前两...
    温暖李花开阅读 161评论 0 0