Leetcode 542 01矩阵

Leetcode 542 01矩阵

问题重述
给定一个由 0 和 1 组成的矩阵，找出每个元素到最近的 0 的距离。
两个相邻元素间的距离为 1 。
示例 1：
输入：
[[0,0,0],
[0,1,0],
[0,0,0]]
输出：
[[0,0,0],
[0,1,0],
[0,0,0]]

示例 2：
输入：
[[0,0,0],
[0,1,0],
[1,1,1]]
输出：
[[0,0,0],
[0,1,0],
[1,2,1]]

思路
这一题我们用dp数组进行动态搜索，先从左上到右下，再从右下到左上，这样就对四个方向都进行了搜索。这样做可以大大减小搜素时间。

最终代码

class Solution {
public:
    vector<vector<int>> updateMatrix(vector<vector<int>>& matrix) {
        if (matrix.empty()) {
            return {};
        }
        int n = matrix.size(), m = matrix[0].size();
        vector<vector<int>> dp(n, vector<int>(m, INT_MAX - 1));
        for (int i = 0; i < n; ++i) {
            for (int j = 0; j < m; ++j) {
                if (matrix[i][j] == 0) {
                    dp[i][j] = 0;
                } else {
                    if (i > 0) {
                        dp[i][j] = min(dp[i][j], dp[i-1][j] + 1);
                    } 
                    if (j > 0) {
                        dp[i][j] = min(dp[i][j], dp[i][j-1] + 1);
                    }
                }
            }
        }
        for (int i = n - 1; i >= 0; --i) {
            for (int j = m - 1; j >= 0; --j) {
                if (matrix[i][j] == 0) {
                    dp[i][j] = 0;
                } else {
                    if (i < n - 1) {
                        dp[i][j] = min(dp[i][j], dp[i+1][j] + 1);
                    } 
                    if (j < m - 1) {
                        dp[i][j] = min(dp[i][j], dp[i][j+1] + 1);
                    }
                }
            }
        }
        return dp;
        }
};