题目
一个机器人位于一个 m x n 网格的左上角 (起始点在下图中标记为“Start” )。
机器人每次只能向下或者向右移动一步。机器人试图达到网格的右下角(在下图中标记为“Finish”)。
现在考虑网格中有障碍物。那么从左上角到右下角将会有多少条不同的路径?
网格中的障碍物和空位置分别用 1 和 0 来表示。
示例 1:
输入:obstacleGrid = [[0,0,0],[0,1,0],[0,0,0]]
输出:2
解释:
3x3 网格的正中间有一个障碍物。
从左上角到右下角一共有 2 条不同的路径:
- 向右 -> 向右 -> 向下 -> 向下
- 向下 -> 向下 -> 向右 -> 向右
示例 2:
输入:obstacleGrid = [[0,1],[0,0]]
输出:1
- 方法一:DFS+缓存->备忘录法
class Solution {
public:
int uniquePathsWithObstacles(vector<vector<int>>& obstacleGrid) {
int m=obstacleGrid.size();
int n=obstacleGrid[0].size();
cache = vector<vector<int>>(m,vector<int>(n));
//初始化起点位置
cache[0][0] = obstacleGrid[0][0]==1 ? 0 :1;
return dfs(obstacleGrid, m-1, n-1);
}
private:
vector<vector<int>> cache;
int dfs(vector<vector<int>>& obstacleGrid, int x, int y){
if(x<0 || y<0) return 0;//范围越界
if(obstacleGrid[x][y]==1) return 0;//遇到障碍物
if(x==0 && y==0) return cache[0][0];//到达起点
if(cache[x][y]>0) return cache[x][y];//该节点已经求出
else return cache[x][y] = dfs(obstacleGrid,x-1,y) + dfs(obstacleGrid,x,y-1);//更新cache
}
};
-
时间复杂度O(n^2)
-
空间复杂度O(n^2)
-
方法二:动态规划
class Solution {
public:
int uniquePathsWithObstacles(vector<vector<int>>& obstacleGrid) {
int m=obstacleGrid.size();//记录行数
int n=obstacleGrid[0].size();//记录列数
vector<vector<int>> dp(m,vector<int>(n));//建立dp数组
dp[0][0] = obstacleGrid[0][0]==1? 0:1;//初始化dp[0][0]防止后续初始化行和列时越界
//初始化第一行
for(int i=1; i<n; ++i){
dp[0][i] = obstacleGrid[0][i]==1 ? 0:(dp[0][i-1]==0?0:1);
}
//初始化第一列
for(int j=1; j<m; ++j){
dp[j][0] = obstacleGrid[j][0]==1 ? 0:(dp[j-1][0]==0?0:1);
}
//计算其余位置
for(int i=1; i<m; ++i){
for(int j=1; j<n; ++j){
dp[i][j] = obstacleGrid[i][j]==1 ? 0 : dp[i-1][j]+dp[i][j-1];
}
}
return dp[m-1][n-1];//返回终点处的
}
};
------------------优化空间复杂度到O(n)----------------------
class Solution {
public:
int uniquePathsWithObstacles(vector<vector<int>>& obstacleGrid) {
int m=obstacleGrid.size();
int n=obstacleGrid[0].size();
vector<int> dp(n);
//相当于初始化dp[0][0]
dp[0] = obstacleGrid[0][0]==1 ? 0:1;
for(int i=0; i<m; ++i){
dp[0] = dp[0]==0 ? 0 : (obstacleGrid[i][0] ? 0 : 1);//相当于初始化第一列
for(int j=1; j<n; ++j){
dp[j] = obstacleGrid[i][j]==1 ? 0 : dp[j]+dp[j-1];
}
}
return dp[n-1];
}
};
- 时间复杂度O(n^2)
- 空间复杂度O(n^2)可以优化到O(n)
- 思路
- 对第一行和第一列而言,如果有一个障碍物,则往后的所有结果都是0
- 对其余节点而言,如果该位置有障碍物,则为0,否则为其上面的节点与左边的节点的路径之和。