leetcode 63. 不同路径 Ⅱ
题意
一个机器人位于一个 m x n 网格的左上角 (起始点在下图中标记为“Start” )。
机器人每次只能向下或者向右移动一步。机器人试图达到网格的右下角(在下图中标记为“Finish”)。
现在考虑网格中有障碍物。那么从左上角到右下角将会有多少条不同的路径?
网格中的障碍物和空位置分别用 1 和 0 来表示。
说明:m 和 n 的值均不超过 100。
示例 1:
输入:
[
[0,0,0],
[0,1,0],
[0,0,0]
]
输出: 2
解释:
3x3 网格的正中间有一个障碍物。
从左上角到右下角一共有 2 条不同的路径:
- 向右 -> 向右 -> 向下 -> 向下
- 向下 -> 向下 -> 向右 -> 向右
解题思路
使用动态规划的思想。
为了方便操作和计算,我在遍历矩阵的时候,把代表障碍的1全部换成 Integer.MAX_VALUE
。
- 我们可以容易确定和起始位置同一行或同一列的所有网格的可能路径数均为1(在没有障碍的情况下)。如果有障碍,那么障碍前的网格均为1,障碍后的网格均为0。
- 我们知道到达任意一个网格
obstacleGrid[i][j]
的可能路径数是该网格上边的网格obstacleGrid[i-1][j]
的可能路径数加上该网格左边的网格obstacleGrid[i][j-1]
的可能路径数,因此得到递推式obstacleGrid[i][j] = obstacleGrid[i-1][j] + obstacleGrid[i][j - 1]
。 - 在递推式中,我们需要根据
obstacleGrid[i-1][j]
和obstacleGrid[i][j - 1]
是否为障碍,即是否等于Integer.MAX_VALUE
,来进行取舍。
代码
class Solution {
public int uniquePathsWithObstacles(int[][] obstacleGrid) {
int row = obstacleGrid.length;
int col = obstacleGrid[0].length;
if (obstacleGrid[0][0] == 1 || obstacleGrid[row - 1][col - 1] == 1)
{
return 0;
}
// 判断只有一行或一列的情况
if (row == 1)
{
for (int j = 0; j < col; j++)
{
if (obstacleGrid[0][j] == 1)
{
return 0;
}
}
return 1;
}
if (col == 1)
{
for (int i = 0; i < row; i++)
{
if (obstacleGrid[i][0] == 1)
{
return 0;
}
}
return 1;
}
// 如果第一行或第一列有障碍,保存该障碍的位置
int rowLimit = row;
int colLimit = col;
for (int i = 0; i < row; i++)
{
for (int j = 0; j < col; j++)
{
if (obstacleGrid[i][j] == 1)
{
if (i == 0)
{
colLimit = j;
}
if (j == 0)
{
rowLimit = i;
}
obstacleGrid[i][j] = Integer.MAX_VALUE;
continue;
}
if ((i == 0 && j < colLimit) || (i < rowLimit && j == 0))
{
obstacleGrid[i][j] = 1;
}
else if (i > 0 && j > 0)
{
if (obstacleGrid[i - 1][j] == Integer.MAX_VALUE && obstacleGrid[i][j - 1] != Integer.MAX_VALUE)
{
obstacleGrid[i][j] = obstacleGrid[i][j - 1];
continue;
}
if (obstacleGrid[i][j - 1] == Integer.MAX_VALUE && obstacleGrid[i - 1][j] != Integer.MAX_VALUE)
{
obstacleGrid[i][j] = obstacleGrid[i - 1][j];
continue;
}
if (obstacleGrid[i][j - 1] == Integer.MAX_VALUE && obstacleGrid[i - 1][j] == Integer.MAX_VALUE)
{
obstacleGrid[i][j] = 0;
continue;
}
obstacleGrid[i][j] = obstacleGrid[i][j - 1] + obstacleGrid[i - 1][j];
}
}
}
return obstacleGrid[row - 1][col - 1];
}
}