leetcode之动态规划

仅以此博客作为刷题的学习笔记。
题目内容如下:

给定一个由 0 和 1 组成的矩阵 mat ，请输出一个大小相同的矩阵，其中每一个格子是 mat 中对应位置元素到最近的 0 的距离。

如图所示：

输入：mat = [[0,0,0],[0,1,0],[1,1,1]]
输出：[[0,0,0],[0,1,0],[1,2,1]]

看到题目就知道可以用动态规划来解决问题，但是在处理矩阵边界位置时(如矩阵中第一行第一列的元素)，总是报错(下标超范围)。

后来看官方的解题思路，才发现对于矩阵边界问题，直接用两次两层for循环就可以解决边界问题：第一次从左上角循环到右下角，来解决矩阵最后一行和最后一列的边界数值；第二次从右下角循环到左上角，来解决矩阵第一行和第一列的边界数值。代码如下：

class Solution(object):
    def updateMatrix(self, mat):
        result = [[100000] * len(mat[0]) for _ in range(len(mat))]
        ##让值为0的位置距离为0
        for i in range(len(mat)):
            for j in range(len(mat[0])):
                if mat[i][j] == 0:
                    result[i][j] = 0
        ##从左上到右下扫描
        for i in range(len(result)):
            for j in range(len(result[0])):
                if i-1>=0:
                    result[i][j]=min(result[i][j],result[i-1][j]+1)
                if j-1>=0:
                    result[i][j]=min(result[i][j],result[i][j-1]+1)
        ##从右下到左上扫描
        for i in range(len(mat)-1,-1,-1):
            for j in range(len(mat[0])-1,-1,-1):
                if i+1<=len(mat)-1:
                    result[i][j]=min(result[i][j],result[i+1][j]+1)
                if j+1<=len(mat[0])-1:
                    result[i][j]=min(result[i][j],result[i][j+1]+1)
        return result