-
Notifications
You must be signed in to change notification settings - Fork 613
/
63.py
41 lines (33 loc) · 1.33 KB
/
63.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
'''
A robot is located at the top-left corner of a m x n grid (marked 'Start' in the diagram below).
The robot can only move either down or right at any point in time. The robot is trying to reach the bottom-right corner of the grid (marked 'Finish' in the diagram below).
Now consider if some obstacles are added to the grids. How many unique paths would there be?
'''
class Solution(object):
def uniquePathsWithObstacles(self, obstacleGrid):
"""
:type obstacleGrid: List[List[int]]
:rtype: int
"""
m, n = len(obstacleGrid), len(obstacleGrid[0])
dp = [[0 for _ in range(n)] for _ in range(m)]
if obstacleGrid[0][0] == 1 or obstacleGrid[m-1][n-1] == 1:
return 0
dp[0][0] = 1
for index in range(1, m):
if obstacleGrid[index][0] == 1:
dp[index][0] = 0
else:
dp[index][0] = dp[index-1][0]
for index in range(1, n):
if obstacleGrid[0][index] == 1:
dp[0][index] = 0
else:
dp[0][index] = dp[0][index-1]
for index_i in range(1, m):
for index_j in range(1, n):
if obstacleGrid[index_i][index_j] == 1:
dp[index_i][index_j] = 0
else:
dp[index_i][index_j] = dp[index_i-1][index_j] + dp[index_i][index_j-1]
return dp[m-1][n-1]