1. 思路
跟上一个题目相比,添加了路障,其实对于有路障的题目也可以用组合数来做,需要用到组合数+容斥定理,详情可以参见我之前出的一个题目:亡灵峡谷
不过这里因为N,M足够小,所以可以使用动态规划进行求解,在上一个题目的基础上呢,我们只需要将有障碍的dp值为0即可。
2. 代码
class Solution {
public:
int uniquePathsWithObstacles(vector<vector<int>>& obstacleGrid) {
int m = obstacleGrid.size(), n = obstacleGrid[0].size();
vector<long long> dp(n+1, 0);
dp[1] = 1;
for(int i = 1; i <= m; ++i)
{
for(int j = 1; j <= n; ++j)
{
dp[j] = obstacleGrid[i-1][j-1] == 1 ? 0 : dp[j]+dp[j-1];
}
}
return dp[n];
}
};