题目描述:
Follow up for "Unique Paths":
Now consider if some obstacles are added to the grids. How many unique paths would there be?
An obstacle and empty space is marked as 1
and 0
respectively in the grid.
For example,
There is one obstacle in the middle of a 3x3 grid as illustrated below.
[ [0,0,0], [0,1,0], [0,0,0] ]
The total number of unique paths is 2
.
Note: m and n will be at most 100.
我的思路:
紧接着上一道题,需要路径个数,这次使用0,1矩阵表示,遇到1则表示此路不通。
基本思路就是遇到1就表示通过他的路径条数为0,置为0。其他的跟上一道题差不多。
我的代码:
class Solution:
def uniquePathsWithObstacles(self, obstacleGrid):
m, n = len(obstacleGrid), len(obstacleGrid[0])
if obstacleGrid[0][0]:
return 0
for i in range(m):
for j in range(n):
if obstacleGrid[i][j] == 1:
obstacleGrid[i][j] = 0
else:
if i == 0 and j == 0:
obstacleGrid[i][j] = 1
elif i == 0:
obstacleGrid[i][j] = obstacleGrid[i][j-1]
elif j == 0:
obstacleGrid[i][j] = obstacleGrid[i-1][j]
else:
obstacleGrid[i][j] = obstacleGrid[i - 1][j] + obstacleGrid[i][j - 1]
return obstacleGrid[-1][-1]
Discuss:
public int uniquePathsWithObstacles(int[][] obstacleGrid) {
int width = obstacleGrid[0].length;
int[] dp = new int[width];
dp[0] = 1;
for (int[] row : obstacleGrid) {
for (int j = 0; j < width; j++) {
if (row[j] == 1)
dp[j] = 0;
else if (j > 0)
dp[j] += dp[j - 1];
}
}
return dp[width - 1];
}
学到:
Discuss中的这个办法也不错,也跟上道题类似,只不过是一行循环处理,一个道理。空间复杂度降低,但是我的代码是O(1)的空间复杂度。