python:迷路的机器人

设想有个机器人坐在一个网格的左上角，网格 r 行 c 列。机器人只能向下或向右移动，但不能走到一些被禁止的网格（有障碍物）。设计一种算法，寻找机器人从左上角移动到右下角的路径。网格中的障碍物和空位置分别用 1 和 0 来表示。
返回一条可行的路径，路径由经过的网格的行号和列号组成。左上角为 0 行 0 列。如果没有可行的路径，返回空数组。
示例 1:
输入:
[
  [0,0,0],
  [0,1,0],
  [0,0,0]
]
输出: [[0,0],[0,1],[0,2],[1,2],[2,2]]
解释: 
输入中标粗的位置即为输出表示的路径，即
0行0列（左上角） -> 0行1列 -> 0行2列 -> 1行2列 -> 2行2列（右下角）
说明：r 和 c 的值均不超过 100。

思路：
1、判断不能到达的点的条件

• 该点为障碍物
• 该点的上面的点和左边的点都不能到达

2、用数组存放该点是否可以到达
3、从终点倒推路径再翻转

class Solution:
    def pathWithObstacles(self, obstacleGrid: List[List[int]]) -> List[List[int]]:
        ans = []
        #行数和列数
        m = len(obstacleGrid)
        n = len(obstacleGrid[0])
        #如果行或列为0或起点或者终点是障碍物，直接返回ans
        if m == 0 or n == 0:
            return ans
        if obstacleGrid[0][0] or obstacleGrid[m-1][n-1]:
            return ans
        #初始化数组，1为可到达，0为不可到达
        mn = [[1 for _ in range(n)] for _ in range(m)]
        #更新首行点是否能到达
        for i in range(1,n):
            if obstacleGrid[0][i] == 1:
                mn[0][i] = 0
            else:
                mn[0][i] = mn[0][i-1]
        #更新首列点是否能到达
        for i in range(1,m):
            if obstacleGrid[i][0] == 1:
                mn[i][0] = 0
            else:
                mn[i][0] = mn[i-1][0]
        #更新中间的点
        for i in range(1,m):
            for j in range(1,n):
                if obstacleGrid[i][j] == 1:
                    mn[i][j] = 0
                else:
                    mn[i][j] = max(mn[i-1][j],mn[i][j-1])
        if mn[-1][-1] == 0:
            return ans
        #倒推路径
        i = m-1
        j = n-1
        while i or j:
            ans.append([i,j])
            if j == 0:
                i -= 1
            elif i == 0:
                j -= 1
            else:
                if mn[i-1][j]>=mn[i][j-1]:
                    i -= 1
                else:
                    j -= 1
        ans.append([0,0])
        ans.reverse()
        return ans