题目
一个机器人位于一个 m × n m \times n m×n 网格的左上角 (起始点在下图中标记为“Start” )。
机器人每次只能向下或者向右移动一步。机器人试图达到网格的右下角(在下图中标记为“Finish”)。
现在考虑网格中有障碍物。那么从左上角到右下角将会有多少条不同的路径?
网格中的障碍物和空位置分别用 1
和 0
来表示。
说明:m 和 n 的值均不超过 100。
示例 1:
输入:
[
[0,0,0],
[0,1,0],
[0,0,0]
]
输出: 2
解释:
3x3 网格的正中间有一个障碍物。
从左上角到右下角一共有 2 条不同的路径:
1. 向右 -> 向右 -> 向下 -> 向下
2. 向下 -> 向下 -> 向右 -> 向右
思路
同上一题 62. 不同路径 一样,还是用递推的思路,当遍历到障碍物时,这个网格到终点的不同路径数直接等于0,不同的是这里有主注意的和优化的:
- 障碍物可能是终点,也可能是起点,需要考虑到这种情况;
- 记录每个格子到终点的路径数量可直接放在
obstacleGrid
数组中,障碍物的格子直接赋值为0;
代码
public class Solution {
public int uniquePathsWithObstacles(int[][] obstacleGrid) {
int m = obstacleGrid.length, n = 0;
if (m > 0)
n = obstacleGrid[0].length;
obstacleGrid[m-1][n-1] = obstacleGrid[m-1][n-1]^1;
for (int j=m-2; j>=0; j--) {
if (obstacleGrid[j][n - 1] != 1)
obstacleGrid[j][n - 1] = obstacleGrid[j + 1][n - 1];
else
obstacleGrid[j][n - 1] = 0;
}
for (int i=n-2; i>=0; i--){
for (int j=m-1; j>=0; j--){
if (obstacleGrid[j][i] == 1){
obstacleGrid[j][i] = 0;
continue;
}
if (i+1 < n)
obstacleGrid[j][i] += obstacleGrid[j][i+1];
if (j+1 < m)
obstacleGrid[j][i] += obstacleGrid[j+1][i];
}
}
return obstacleGrid[0][0];
}
public static void main(String[] args) {
Solution solution = new Solution();
int[][] a = {{1},{0}};
System.out.println(solution.uniquePathsWithObstacles(a));
}
}