力扣—63不同路径II回溯解法

题目描述

一个机器人位于一个 m x n 网格的左上角 （起始点在下图中标记为“Start” ）。

机器人每次只能向下或者向右移动一步。机器人试图达到网格的右下角（在下图中标记为“Finish”）。

现在考虑网格中有障碍物。那么从左上角到右下角将会有多少条不同的路径？

网格中的障碍物和空位置分别用 1 和 0 来表示。

示例 1:

输入:
[
  [0,0,0],
  [0,1,0],
  [0,0,0]
]
输出: 2
解释:
3x3 网格的正中间有一个障碍物。
从左上角到右下角一共有 2 条不同的路径：
1. 向右 -> 向右 -> 向下 -> 向下
2. 向下 -> 向下 -> 向右 -> 向右

解题思路

采用回溯的解题思路，首先起点是（0， 0）处，只能向下或者向右移动才能到target（不可能向上或者向左，可以想象如果先向下走一步，然后又朝上走一步，那么就永远走不出来了），有几个递归终止条件，首先是坐标超出了范围，即到达了最下边或者最右边，另一个是如果当前走的遇到了障碍物，那么就返回0。直到走到了最右边才能算一条路，即返回1

class Solution(object):
    def __init__(self):
        self._dict = {}
    def uniquePathsWithObstacles(self, obstacleGrid):
        """
        :type obstacleGrid: List[List[int]]
        :rtype: int
        """
        m = len(obstacleGrid); n = len(obstacleGrid[0])
        def dfs(x, y, matrix):
            if x >= m or y >= n or matrix[x][y] == 1:
                return 0
            if x == m - 1 and y == n - 1:
                return 1
            if (x, y) in self._dict:
                return self._dict[(x, y)]
            self._dict[(x, y)] = dfs(x, y + 1, matrix) + dfs(x + 1, y, matrix)
            return self._dict[(x, y)]
        res = dfs(0, 0, obstacleGrid)
        return res
        
S = Solution()
print(S.uniquePathsWithObstacles([[0,0,0],[0,1,0],[0,0,0]]))