题目:一个机器人位于一个 m * n 网格的左上角 (起始点在下图中标记为 “Start” )。
机器人每次只能向下或者向右移动一步。机器人试图达到网格的右下角(在下图中标记为 “Finish” )。问总共有多少条不同的路径?但是格子中可能有障碍,网格中的障碍物和空位置分别用 1 和 0 来表示。
示例 1:
输入:obstacleGrid = [[0,0,0],[0,1,0],[0,0,0]]
输出:2
解释:
3x3 网格的正中间有一个障碍物。
从左上角到右下角一共有 2 条不同的路径:
- 向右 -> 向右 -> 向下 -> 向下
- 向下 -> 向下 -> 向右 -> 向右
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/unique-paths-ii
思路:新建一个dp[][]二维数组保存到达每个位置有多少不同方式。
如果没有障碍物第一排,第一列均只有一种方式到达;其余位置都是左边和上边位置的和。
但是有障碍物的话,第一排的障碍物和它之后的位置都不可达,第一列同理(在for循环里加条件obstacleGrid[i][0] == 0,表示不是障碍物才有一种走法,否则就是障碍物,连同它后面的位置不处理,默认是0种方式),其余位置如果不是障碍物obstacleGrid[i][j] == 0依旧是左边和上边位置的和,是障碍物就不处理,默认0种
class Solution {
public int uniquePathsWithObstacles(int[][] obstacleGrid) {
if (obstacleGrid == null || obstacleGrid.length == 0) {
return 0;
}
// 定义 dp 数组并初始化第 1 行和第 1 列。
int m = obstacleGrid.length, n = obstacleGrid[0].length;
int[][] dp = new int[m][n];
// 如果列有一个障碍,那这障碍下面都走不了 条件obstacleGrid[i][0] == 0
for (int i = 0; i < m && obstacleGrid[i][0] == 0; i++) {
dp[i][0] = 1;
}
for (int j = 0; j < n && obstacleGrid[0][j] == 0; j++) {
dp[0][j] = 1;
}
// 根据状态转移方程 dp[i][j] = dp[i - 1][j] + dp[i][j - 1] 进行递推。
for (int i = 1; i < m; i++) {
for (int j = 1; j < n; j++) {
//其余位置如果不是障碍物依旧是左边和上边位置的和,是障碍物就不处理,默认0种
if (obstacleGrid[i][j] == 0) {
dp[i][j] = dp[i - 1][j] + dp[i][j - 1];
}
}
}
return dp[m - 1][n - 1];
}