描述
“不同的路径” 的跟进问题:
现在考虑网格中有障碍物,那样将会有多少条不同的路径?
网格中的障碍和空位置分别用 1 和 0 来表示。
注意事项
m 和 n 均不超过100
样例
[
[0,0,0],
[0,1,0],
[0,0,0]
]
其中,一共有2条不同的路径从左上角到右下角。
思考
- 相比 不同的路径I 多了一个障碍的条件,即不对有障碍物的点进行操作
- 由于数组下标长宽弄错,导致下标越界,折腾了半个小时,还是在 VS 里面断点一步一步找出来的,希望大家也可以多注意点
代码
// By Lentitude
class Solution {
public:
/**
* @param obstacleGrid: A list of lists of integers
* @return: An integer
*/
int uniquePathsWithObstacles(vector<vector<int> > &obstacleGrid) {
// write your code here
vector<vector<int> > vec(obstacleGrid);
int height = vec.size();
int width = vec[0].size();
// 把 0 全部转化为 1, 1 转化为0,便于后面直接对 1 和0分开操作
for (int i = 0; i != height; ++i) {
for (int j = 0; j != width; ++j) {
if (vec[i][j] == 1) {
vec[i][j] = 0;
}
else {
vec[i][j] = 1;
}
}
}
// 将第一行中 101 -> 100
for (int i = 0; i != width; ++i) {
if (vec[0][i] == 0) {
while (i < width) {
vec[0][i++] = 0;
}
break;
}
}
/**
* 将第一列中 1 1
* 0 -> 0
* 1 0
*/
for (int i = 0; i != height; ++i) {
if (vec[i][0] == 0) {
while (i < height) {
vec[i++][0] = 0;
}
break;
}
}
// 如果是单行或者单列,则不需要进行运算,可以直接返回结果
if (height == 1 || width == 1) {
return vec[height - 1][width - 1];
}
// 否则, 需要进行运算
for (int i = 1; i != height; ++i) {
for (int j = 1; j != width; ++j) {
if (vec[i][j] != 0) {
vec[i][j] = vec[i - 1][j] + vec[i][j - 1];
}
}
}
return vec[height - 1][width - 1];
}
};