原题:https://leetcode-cn.com/problems/unique-paths-ii/
思路
与上题相同,不过是加了路障。地图上每一格都有两个状态,有路障或无路障,分别以1和0表示其状态。
每个格还有到此格的路径数值。无路障的格路径数为上方格路径数加左方格路径数。有路障的格路径数为0。
因此可用两个矩阵(可以合并为一个)来表示整张地图的路径数。
代码
class Solution:
def uniquePathsWithObstacles(self, obstacleGrid):
m, n = len(obstacleGrid[0]), len(obstacleGrid)
values = [[0]*m for j in range(n)]
for i in range(m):
if obstacleGrid[0][i] == 0:
values[0][i] = 1
else:
break
for j in range(n):
if obstacleGrid[j][0] == 0:
values[j][0] = 1
else:
break
for j in range(1, n):
for i in range(1, m):
if obstacleGrid[j][i] == 1:
values[j][i] = 0
else:
values[j][i] = values[j-1][i] + values[j][i-1]
return values[n-1][m-1]