class Solution {
public:
int uniquePathsWithObstacles(vector<vector<int>>& obstacleGrid) {
int m=obstacleGrid.size();
int n=obstacleGrid[0].size();
if(obstacleGrid[0][0]||obstacleGrid[m-1][n-1])
return 0;
vector<int> a(n,0);
vector<vector<int> > dp;
for(int i=0;i<m;i++)
dp.push_back(a);
dp[0][0]=1;
for(int i=1;i<m;i++)
dp[i][0]=obstacleGrid[i][0]==0?dp[i-1][0]:0;
for(int i=1;i<n;i++)
dp[0][i]=obstacleGrid[0][i]==0?dp[0][i-1]:0;
for(int i=1;i<m;i++)
for(int j=1;j<n;j++)
{
if(!obstacleGrid[i][j])
dp[i][j]=dp[i-1][j]+dp[i][j-1];
}
return dp[m-1][n-1];
}
};
Updated version which only take O(n) space.
class Solution {
public:
int uniquePathsWithObstacles(vector<vector<int>>& obstacleGrid) {
int m=obstacleGrid.size();
if(m==0)
return 0;
int n=obstacleGrid[0].size();
int dp[n];
dp[0]=obstacleGrid[0][0]==0?1:0;
for(int i=1;i<n;i++)
dp[i]=obstacleGrid[0][i]==0?dp[i-1]:0;
for(int i=1;i<m;i++)
{
if(obstacleGrid[i][0]==1)
dp[0]=0;
for(int j=1;j<n;j++)
dp[j]=obstacleGrid[i][j]==0?dp[j-1]+dp[j]:0;
}
return dp[n-1];
}
};