Leetcode 542 Matrix

最新推荐文章于 2022-10-09 17:46:36 发布

TheOneAc1

最新推荐文章于 2022-10-09 17:46:36 发布

阅读量393

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/TheOneAc1/article/details/78175689

leetcode 专栏收录该内容

1 篇文章

订阅专栏

本文探讨了两种更新矩阵中每个元素到最近0的距离的方法：一种使用BFS遍历算法，另一种采用动态规划方法。通过对比，展示了动态规划在解决该问题上的效率优势。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接

第一版BFS：没找到毛病通不过，也是醉了（复杂度有点高呀，小伙子）


class Solution {
public:
   void bfs(vector<vector<int>>&matrix,int i, int j, int &cnt){
        if(matrix[i][j] == 0)return;
        cnt++;
        if(i > 0)bfs(matrix, i-1,j,cnt);
        else if(j > 0)bfs(matrix, i,j-1,cnt);
        else if(i < matrix.size() -1)bfs(matrix, i +1, j,cnt);
        else if(j < matrix[0].size() -1)bfs(matrix, i, j + 1,cnt);
    }
    vector<vector<int>> updateMatrix(vector<vector<int>>& matrix) {
        vector<vector<int>> res(matrix.size(),vector<int>(matrix[0].size(),0));
        for(int i = 0; i< matrix.size(); i++){
            for(int j = 0; j < matrix[0].size(); j++){
                int cnt = 0;
                bfs(matrix,i,j, cnt);
                res[i][j] = cnt;
            }
        }
        return res;
    }
};

第二版DP，over

class Solution {
public:

    vector<vector<int>> updateMatrix(vector<vector<int>>& matrix) {
        vector<vector<int>> res(matrix.size(),vector<int>(matrix[0].size(),INT_MAX));
        for(int i = 0; i< matrix.size(); i++){
            for(int j = 0; j < matrix[0].size(); j++){
                if(matrix[i][j] == 0)res[i][j]=0;
                if(i > 0 && res[i-1][j] != INT_MAX)res[i][j] = min(res[i][j],res[i-1][j]+1);
                if(j > 0 && res[i][j-1] != INT_MAX)res[i][j] = min(res[i][j],res[i][j-1]+1);

            }
        }
        for(int i = matrix.size()-1; i>=0; i--){
            for(int j = matrix[0].size()-1; j >=0; j--){
                if(matrix[i][j] == 0)res[i][j]=0;
                if(i < matrix.size()-1 && res[i+1][j]!= INT_MAX)res[i][j] = min(res[i][j],res[i+1][j]+1);
                if(j < matrix[0].size()-1  && res[i][j+1] != INT_MAX )res[i][j] = min(res[i][j],res[i][j+1]+1);

            }
        }
        return res;
    }
};