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> updateMatrix(vector>& mat) {
        int m = mat.size(),n = mat[0].size();
        vector> res(m,vector(n,INT_MAX-1));
        // 第一次从左上到右下遍历
        for(int i=0;i0)
                        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;

    }
};