不同路径 II
- 一个机器人位于一个 m x n 网格的左上角 (起始点在下图中标记为 “Start” )。
机器人每次只能向下或者向右移动一步。机器人试图达到网格的右下角(在下图中标记为 “Finish”)。
现在考虑网格中有障碍物。那么从左上角到右下角将会有多少条不同的路径?
网格中的障碍物和空位置分别用 1 和 0 来表示。
示例 1:
输入:obstacleGrid = [[0,0,0],[0,1,0],[0,0,0]]
输出:2
解释:3x3 网格的正中间有一个障碍物。
从左上角到右下角一共有 2 条不同的路径:
- 向右 -> 向右 -> 向下 -> 向下
- 向下 -> 向下 -> 向右 -> 向右
解题思路
动态规划思路
- 定义状态:
用一个二维数组 dp 来保存到达每个位置的路径数量。
dp[i][j] 表示从起点到位置 (i, j) 的路径数量。 - 初始化:
如果起点 (0, 0) 是障碍物,则直接返回 0,因为无法出发。
初始化起点 dp[0][0] = 1,表示机器人从起点开始。 - 状态转移:
对于每个位置 (i, j),如果是障碍物,dp[i][j] = 0。
否则,dp[i][j] 等于从上方到达该位置的路径数加上从左侧到达该位置的路径数,即 dp[i][j] = dp[i-1][j] + dp[i][j-1]。 - 结果:
dp[m-1][n-1] 即为从起点到终点的路径数量。
Java实现
public class UniquePathsII {
public int uniquePathsWithObstacles(int[][] obstacleGrid) {
int m = obstacleGrid.length;
int n = obstacleGrid[0].length;
// 如果起点是障碍物,则无法出发
if (obstacleGrid[0][0] == 1) {
return 0;
}
// 初始化 dp 数组
int[][] dp = new int[m][n];
// 初始化起点
dp[0][0] = 1;
// 填充 dp 数组
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
// 障碍物填充
if (obstacleGrid[i][j] == 1) {
dp[i][j] = 0; // 障碍物位置的路径数为 0
} else {
if (i > 0) {
dp[i][j] += dp[i-1][j]; // 从上方到达
}
if (j > 0) {
dp[i][j] += dp[i][j-1]; // 从左侧到达
}
}
}
}
// 返回终点的路径数
return dp[m-1][n-1];
}
// 测试用例
public static void main(String[] args) {
UniquePathsII solution = new UniquePathsII();
int[][] obstacleGrid = {
{0, 0, 0},
{0, 1, 0},
{0, 0, 0}
};
System.out.println(solution.uniquePathsWithObstacles(obstacleGrid)); // 期望输出: 2
}
}
时间空间复杂度
- 时间复杂度:O(m * n),其中 m 是网格的行数,n 是网格的列数,需要遍历整个网格。
- 空间复杂度:O(m * n),需要一个同样大小的 dp 数组来存储每个位置的路径数量。