这道题是Unique Paths的升级版,难度略有上升但还是可以接受的,下面就和大家分享一下经验吧!
题目如下:
题意分析:
给定一个m*n网格,让一个机器人从左上角每次向右或者向下移动一步最终到达右下角,有障碍物的地方是无法通过的,其中1表示有障碍物,0表示无障碍物,请问有多少条这样的路径?
方法一(动态规划法)
维护一个大小为 (m+1) x (n+1)的二维 dp 数组(dp 数组的大小设为 (m+1) x (n+1) 是为了处理第一行(i = 0)和和第一列 ( j =0 ) 的特殊情况,当 i或 j 为0时,减1避免出错),其中 dp[i][j] 表示到达 (i-1, j-1) 位置的不同路径的个数。由于每个位置都可以由上方和左方而来,所以转移方程可归纳为: dp[i][j] = dp[i - 1][j] + dp[i][j - 1]。
当某个位置是障碍物时直接跳过该位置,并保持其dp值为0。当起点是障碍物时程序直接返回 0,否则dp[1][1] = 1,由于 dp[1][1] 是由 dp[0][1] + dp[1][0] 更新而来,所以需要将二者中任意一个初始化为1即可。经测试使用int型的 dp 数组会产生溢出的错误,所以必须改为使用 long 型的数组来避免溢出。
解题代码如下:
class Solution{
public:
int uniquePathsWithObstacles(vector<vector<int>>& obstacleGrid){
int row = obstacleGrid.size();
if ( row == 0 || obstacleGrid[0][0] == 1 ) return 0;
int col = obstacleGrid[0].size();
vector<vector<long>> dp( row + 1, vector<long> ( col+1, 0) );
dp[0][1] = 1;
for (int i = 1; i <= row; i++) {
for (int j = 1; j <= col; j++) {
//如果有障碍物,则不用更新dp,只需保存为0即可
if ( obstacleGrid[i-1][j-1] != 0 ) continue;
dp[i][j] = dp[i-1][j] + dp[i][j-1];
}
}
return dp[row][col];
}
};
提交后的结果如下:
方法二(优化方法一)
其实可以只维护一维dp数组即可求解,减少了空间复杂度,具体算法细节请参考代码与注释。
解题代码如下:
class Solution{
public:
int uniquePathsWithObstacles(vector<vector<int>>& obstacleGrid){
int row = obstacleGrid.size();
if ( row == 0 || obstacleGrid[0][0] == 1 ) return 0;
int col = obstacleGrid[0].size();
vector<long> dp( col, 0 );
dp[0] = 1;
//从第一行开始考虑,由于第一行只有从左边来的即只加左边的dp[j-1],所以初始化时dp数组均为0
for (int i = 0; i < row; i++) {
//对于第一列的位置,由于第一列只有从上边来,所以无须累加操作,直接用上一层的结果即可
for (int j = 0; j < col; j++) {
//障碍物处,dp[j]设置为0
if ( obstacleGrid[i][j] == 1 ) dp[j] = 0;
//非障碍物处,用转移方程公式对dp[j]进行更新
else if ( j > 0 ) dp[j] = dp[j] + dp[j-1];
}
}
return dp[col-1];
}
};
提交后的结果如下:
日积月累,与君共进,增增小结,未完待续。