一个机器人位于一个 m x n 网格的左上角 (起始点在下图中标记为“Start” )。
机器人每次只能向下或者向右移动一步。机器人试图达到网格的右下角(在下图中标记为“Finish”)。
现在考虑网格中有障碍物。那么从左上角到右下角将会有多少条不同的路径?
源码来自 代码随想录-63. 不同路径 II
CPP:
class Solution {
public:
int uniquePathsWithObstacles(vector<vector<int>>& obstacleGrid) {
// 获取长宽
int m = obstacleGrid.size();
int n = obstacleGrid[0].size();
// 1. 如果起点和终点是障碍物,直接返回0
if (obstacleGrid[0][0] == 1 || obstacleGrid[m - 1][n - 1] == 1)
return 0;
// 创建dp数组,定义数组类型、数组大小、列的大小,同时给数组所有元素赋值为0。
vector<vector<int>> dp(m, vector<int>(n, 0));
// 2. 初始化,如果碰到障碍物就停下,障碍之后的dp应该还是初始值0。
for (int i = 0; i < m && obstacleGrid[i][0] == 0; i++) dp[i][0] = 1;
for (int j = 0; j < n && obstacleGrid[0][j] == 0; j++) dp[0][j] = 1;
// 开始遍历
for (int i = 1; i < m; i++) {
for (int j = 1; j < n; j++) {
// 3. 如果碰到障碍物就跳过,继续下一层遍历
if (obstacleGrid[i][j] == 1) continue;
// 动规公式
dp[i][j] = dp[i][j - 1] + dp[i - 1][j];
}
}
return dp[m - 1][n - 1];
}
};
javascript代码
var uniquePathsWithObstacles = function(obstacleGrid) {
const m = obstacleGrid.length
const n = obstacleGrid[0].length
// 初始化dp数组
const dp = Array(m).fill().map(item => Array(n).fill(0))
for (let i = 0; i < m && obstacleGrid[i][0] === 0; ++i) {dp[i][0] = 1}
for (let i = 0; i < n && obstacleGrid[0][i] === 0; ++i) {dp[0][i] = 1}
for (let i = 1; i < m; ++i) {
for (let 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]
};