题目描述:
一个机器人位于一个 m x n 网格的左上角 (起始点在下图中标记为“Start” )。
机器人每次只能向下或者向右移动一步。机器人试图达到网格的右下角(在下图中标记为“Finish”)。
现在考虑网格中有障碍物。那么从左上角到右下角将会有多少条不同的路径?
示例:
输入:
obstacleGrid = [[0,0,0],[0,1,0],[0,0,0]]
输出:
2
解释:
3x3 网格的正中间有一个障碍物。
从左上角到右下角一共有 2 条不同的路径:
1. 向右 -> 向右 -> 向下 -> 向下
2. 向下 -> 向下 -> 向右 -> 向右
思路:
和之前的这篇(Leetcode 62 不同路径)一样 ,只不过我们需要在寻路时加上障碍物的判断。
首先,我们需要修改初始化的节点,判断初始化节点是否放置了障碍物。
其次,对原先的寻路函数做如下修改:
最后,需要判断终点是否放置障碍物,若终点放置障碍物,则返回 0。
代码:
class Solution {
public:
int uniquePathsWithObstacles(vector<vector<int>>& obstacleGrid) {
int m = obstacleGrid.size();
int n = obstacleGrid[0].size();
if (m == 1 && n == 1) return obstacleGrid[0][0]?0:1;
int way[100][100] = {0};
//判断初始化节点是否放置了障碍物
way[0][0] = obstacleGrid[0][0]?0:1;
way[0][1] = (n > 1 && obstacleGrid[0][1] || obstacleGrid[0][0])?0:1;
way[1][0] = (m > 1 && obstacleGrid[1][0] || obstacleGrid[0][0])?0:1;
//修改后的寻路方式
for (int i = 0; i < m; ++i){
for (int j = 0; j < n; ++j){
if (i > 0 && j > 0 && !obstacleGrid[i-1][j] && !obstacleGrid[i][j-1])
way[i][j] = way[i-1][j] + way[i][j-1];
else if (i > 0 && j > 0 && obstacleGrid[i-1][j] && !obstacleGrid[i][j-1])
way[i][j] = way[i][j-1];
else if (i > 0 && j > 0 && !obstacleGrid[i-1][j] && obstacleGrid[i][j-1])
way[i][j] = way[i-1][j];
else if (i == 0 && j > 0 && !obstacleGrid[i][j-1])
way[i][j] = way[i][j-1];
else if (j == 0 && i > 0 && !obstacleGrid[i-1][j])
way[i][j] = way[i-1][j];
else
continue;
}
}
//判断终点是否放置了障碍物
if (obstacleGrid[m-1][n-1])
return 0;
else
return way[m-1][n-1];
}
};
结果: