一个机器人位于一个 m x n 网格的左上角 (起始点在下图中标记为“Start” )。
机器人每次只能向下或者向右移动一步。机器人试图达到网格的右下角(在下图中标记为“Finish”)。
现在考虑网格中有障碍物。那么从左上角到右下角将会有多少条不同的路径?
网格中的障碍物和空位置分别用 1 和 0 来表示。
算法思想为设立一个数组
我们用 f(i, j)来表示从坐标(0,0) 到坐标 (i, j)的路径总数
u(i, j)表示坐标 (i, j)是否通行,如果坐标 (i, j)有障碍物,u(i, j) = 0,否则 u(i, j) = 1
拿图来理解一下
此时(2,1)与(1,2)点值都设为了1。因为从出发点到(2,1)(1,2)这两点的路径只有一条,继续往下走
这时(2,2)节点的值应该为2,因为我们限定了机器人只能往下或往右走,因此每次计算下一点的路径只需要计算(i-1,j)与(i,j-1)的值之合,即可得到可以达到此点的路径条数。
如果有障碍物的时候程序应该如何运行呢?
因为我们设有u(i, j)来表示该店是否存在障碍物,若存在时令f(i,j)=0即可,不可到达,也不影响后续路径的判断。
下图为路径计算的函数
代码实现:
class Solution {
public int uniquePathsWithObstacles(int[][] obstacleGrid) {
int n = obstacleGrid.length, m = obstacleGrid[0].length;
int[] f = new int[m];
f[0] = obstacleGrid[0][0] == 0 ? 1 : 0;
for (int i = 0; i < n; ++i) {
for (int j = 0; j < m; ++j) {
if (obstacleGrid[i][j] == 1) {
f[j] = 0;
continue;
}
if (j - 1 >= 0 && obstacleGrid[i][j - 1] == 0) {
f[j] += f[j - 1];
}
}
}
return f[m - 1];
}
}
关于为什么 f 数组只定义了一维
因为从图上我们可以看到 在第一行的时候 如果没有障碍物
第二行的路径计算公式为 f(i-1,j) + f(i,j-1), f(i-1,j) 即为上次我们存储的f(i)的值,因此只需要f[j] += f[j - 1]即相当于f(i-1,j) + f(i,j-1), 节省了空间。