leetcode542 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]]

提示：
给定矩阵的元素个数不超过 10000。
给定矩阵中至少有一个元素是 0。
矩阵中的元素只在四个方向上相邻: 上、下、左、右。
通过次数54,917提交次数120,035

思路：连续两次遍历二维数组，一遍从左向右，从上至下，一遍反之，更新两次与0的距离，从而求出与0的最小值。

代码：

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