题目链接:https://leetcode-cn.com/problems/unique-paths-ii/
题目描述
一个机器人位于一个 m x n 网格的左上角 (起始点在下图中标记为“Start” )。
机器人每次只能向下或者向右移动一步。机器人试图达到网格的右下角(在下图中标记为“Finish”)。
现在考虑网格中有障碍物。那么从左上角到右下角将会有多少条不同的路径?
网格中的障碍物和空位置分别用 1 和 0 来表示。
说明:m 和 n 的值均不超过 100。
示例 1:
输入:
[
[0,0,0],
[0,1,0],
[0,0,0]
]
输出: 2
解释:
3x3 网格的正中间有一个障碍物。
从左上角到右下角一共有 2 条不同的路径:
1. 向右 -> 向右 -> 向下 -> 向下
2. 向下 -> 向下 -> 向右 -> 向右
思路
典型的动态规划题目,同62题解法;我们用一个二维数组dp来存储当前位置的不同路径数。
动态规划方程为:
dp[i][j] = dp[i-1][j] + dp[i][j-1]
区别在于如果obstacleGrid[i][j] == 1
,则直接令dp[i][j]=0
注意:重点在于C++用int必然会越界。怎么改都越界,要强转成(long )
复杂度分析
时间复杂度:O(mn)
空间复杂度:O(mn)
m,n分别为矩阵行列数
class Solution {
public:
int uniquePathsWithObstacles(vector<vector<int>>& obstacleGrid) {
if(obstacleGrid.empty() || obstacleGrid[0].empty()) return 0;
int rows = obstacleGrid.size(), cols = obstacleGrid[0].size();
if (obstacleGrid.back().back() == 1 ||obstacleGrid[0][0] == 1)
return 0;
vector<vector<long >> dp(rows, vector<long >(cols, 0));
dp[0][0] = 1;
for (int i = 0; i < rows; ++i) {
for (int j = 0; j < cols; ++j) {
if(obstacleGrid[i][j] == 1){
dp[i][j] = 0;
continue;
}
if(i==0 && j==0) continue;
long up = 0, left = 0;
if(i-1>=0) up = dp[i-1][j];
if(j-1>=0) left = dp[i][j-1];
dp[i][j] = up+left;
}
}
return dp.back().back();
}
};
进一步优化,我们可以使用obstacleGrid
作为dp数组,将空间复杂度降为O(1)
/*
* 动态规划,使用obstacleGrid数组作为dp数组
* 时间复杂度O(N^2) 空间复杂度O(1)
*/
class Solution {
public:
int uniquePathsWithObstacles(vector<vector<int>>& obstacleGrid) {
if(obstacleGrid.empty() || obstacleGrid[0].empty()) return 0;
int rows = obstacleGrid.size(), cols = obstacleGrid[0].size();
if (obstacleGrid.back().back() == 1 ||obstacleGrid[0][0] == 1)
return 0;
obstacleGrid[0][0] = 1;
// 第一列元素
for (int i = 1; i < rows; ++i)
obstacleGrid[i][0] = (obstacleGrid[i][0] == 0 && obstacleGrid[i-1][0] == 1 ) ? 1:0; // 设置为1表示有1条路径
// 第一行元素
for (int j = 1; j < cols; ++j)
obstacleGrid[0][j] = (obstacleGrid[0][j] == 0 && obstacleGrid[0][j-1] == 1)? 1:0; // 设置为1表示有1条路径
for (int i = 1; i < rows; ++i) {
for (int j = 1; j < cols; ++j) {
if(obstacleGrid[i][j] == 0)
obstacleGrid[i][j] = obstacleGrid[i-1][j] + obstacleGrid[i][j-1];
else
obstacleGrid[i][j] = 0;
}
}
return obstacleGrid.back().back();
}
};