题目
一个机器人位于一个 m x n 网格的左上角 (起始点在下图中标记为“Start” )。
机器人每次只能向下或者向右移动一步。机器人试图达到网格的右下角(在下图中标记为“Finish”)。
现在考虑网格中有障碍物。那么从左上角到右下角将会有多少条不同的路径?
网格中的障碍物和空位置分别用 1 和 0 来表示。
说明:m 和 n 的值均不超过 100。
示例 1:
输入:
[
[0,0,0],
[0,1,0],
[0,0,0]
]
输出: 2
解释:
3x3 网格的正中间有一个障碍物。
从左上角到右下角一共有 2 条不同的路径:
- 向右 -> 向右 -> 向下 -> 向下
- 向下 -> 向下 -> 向右 -> 向右
思路
动态规划
当不考虑障碍物时,到达每个格子的路径数等于上面格子路径数加上左边格子路径数,因为格子只能想右和向下。当考虑障碍物时,将障碍物所在格子的路径数置0即可。
第一步:将第一行第一列所有元素置1,当有障碍物时,将当前格子和其后的格子全置0。
第二步:创建同样大小二维数组,遍历障碍矩阵。 dp[i][j]=dp[i-1][j]+dp[i][j-1];
第三步:返回最后一个格子的值。
代码
class Solution {
public:
int uniquePathsWithObstacles(vector<vector<int>>& obstacleGrid) {
int row = obstacleGrid.size();
if (!row) return 0;
int col = obstacleGrid[0].size();
if (obstacleGrid[row - 1][col - 1]||obstacleGrid[0][0]) return 0;
vector<vector<int>> dp(row, vector<int>(col));
for(int h=0;h<col;h++)
{
if(obstacleGrid[0][h]==1)
{
while(h<col)
{
dp[0][h]=0;
h++;
}
break;
}
else
{
dp[0][h]=1;
}
}
for(int h=0;h<row;h++)
{
if(obstacleGrid[h][0]==1)
{ while(h<row)
{
dp[h][0]=0;
h++;
}
break;
}
else
{
dp[h][0]=1;
}
}
for(int i =1;i<row;i++)
{
for(int j=1;j<col;j++)
{
if(obstacleGrid[i][j]==1)
{
dp[i][j]=0;
continue;
}
else
{
dp[i][j]=dp[i-1][j]+dp[i][j-1];
}
}
}
return dp[row-1][col-1];
}
};