不同路径 II

题目描述

来源于 https://leetcode-cn.com/

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

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

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

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

说明:m 和 n 的值均不超过 100。

示例 1:

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

解法:

只需要在 上一题 的基础上稍加修改即可。

from collections import defaultdict

class Solution:
    def uniquePathsWithObstacles(self, obstacle_grid):
        """
        :type obstacleGrid: List[List[int]]
        :rtype: int
        """
        
        m = len(obstacle_grid)
        n = len(obstacle_grid[0])
        
        dp = defaultdict(int)
        dp[(m-1,n-1)] = 1
        
        for col in range(n-1, -1, -1):
            for row in range(m-1, -1, -1):
                if obstacle_grid[row][col] == 1:
                    dp[(row, col)] = 0
                    continue
                
                if row < m-1:
                    dp[(row, col)] += dp[(row+1, col)]
                
                if col < n-1:
                    dp[(row, col)] += dp[(row, col+1)]
        
        return dp[(0,0)]