題目出處

63. Unique Paths II

難度

Medium

個人範例程式碼

class Solution:
    def uniquePathsWithObstacles(self, obstacleGrid: List[List[int]]) -> int:
        m, n = len(obstacleGrid), len(obstacleGrid[0])

        if not m or not n:
            return 0

        dp = [[0 for _ in range(n)] for _ in range(m)]

        # init
        if obstacleGrid[0][0] == 1:
            return 0
        else:
            dp[0][0] = 1 

        for i in range(1, m):
            if obstacleGrid[i][0] == 1:
                dp[i][0] = 0
            else:
                dp[i][0] = dp[i-1][0] 

        for j in range(1, n):
            if obstacleGrid[0][j] == 1:
                dp[0][j] = 0
            else:
                dp[0][j] = dp[0][j-1]

        # function
        for i in range(1, m):
            for j in range(1, n):
                if obstacleGrid[i][j] == 1:
                    dp[i][j] = 0
                else:
                    dp[i][j] = dp[i-1][j] + dp[i][j-1]

        # print(dp)
        # ans
        return dp[-1][-1]

算法說明

• 此題的前一題為無障礙物版本，可以參考：

【Leetcode】python – [62] Unique Paths 個人解法筆記 #重要題型

也是經典的二維 DP 問題，
透過建立下方的圖片，就可以求得所有路徑的答案

(DP 記錄到達那一格的所有方法數)

而這一題我們只需要多做處理障礙物的部分，但這次我們需要特別小心，

除了中間之外，連「邊邊初始化也要處理」

• 處理中間

• 邊邊初始化 (要多注意)

input handling

如果沒有 m 或 n，return 0 (無方法)
且如果 A[0][0] 位置就是障礙物，也是 return 0 (無方法)

Boundary conditions

用 for 來控制 DP 的範圍

Reference

• 不同的路径 II · Unique Paths II