LeetCode 542.01矩阵

LeetCode 542.01矩阵

ways:

• 第一轮从左上到右下遍历
• 第二轮从右下到左上遍历

证明：
	先明白一个定理：
			如果距离(i,j)最近的点为(i-a,j+b)，那么距离(i,j+b)最近的点也是(i-a,j+b)

	所以：如果第一遍(i,j)没有传到(i-a,j+b)，但是(i-a,j+b)第一遍传到了(i,j+b)，在第二遍的时候，(i,j+b)会传到(i,j)
	所以，结论成立。

完整代码：

class Solution {
public:
    vector<vector<int>> updateMatrix(vector<vector<int>>& mat) {
        int m = mat.size(),n = mat[0].size();
        vector<vector<int>> res(m,vector<int>(n,INT_MAX-1));
        // 第一次从左上到右下遍历
        for(int i=0;i<m;i++)
        {
            for(int j=0;j<n;j++)
            {
                if(mat[i][j] == 0)
                    res[i][j] = 0;
                else
                {
                    if(j>0)
                        res[i][j] = min(res[i][j],res[i][j-1]+1);
                    if(i>0)
                        res[i][j] = min(res[i][j],res[i-1][j]+1);
                }
            }
        }
        // 第二次从右下到左上遍历
        for(int i=m-1;i>=0;i--)
        {
            for(int j=n-1;j>=0;j--)
            {
                if(mat[i][j] != 0)
                {
                    if(i+1 < m)
                        res[i][j] = min(res[i][j],res[i+1][j]+1);
                    if(j+1 < n)
                        res[i][j] = min(res[i][j],res[i][j+1]+1);
                }
            }
        }
        return res;

    }
};