题目:
Follow up for “Unique Paths”:
Now consider if some obstacles are added to the grids. How many unique paths would there be?
An obstacle and empty space is marked as1and0respectively in the grid.
For example,
There is one obstacle in the middle of a 3x3 grid as illustrated below.
[
[0,0,0],
[0,1,0],
[0,0,0]
]
The total number of unique paths is2.
Note: m and n will be at most 100.
程序:
class Solution {
public:
int uniquePathsWithObstacles(vector<vector<int> > &obstacleGrid) {
int n = obstacleGrid.size();
int m = obstacleGrid[0].size();
if(n==0 || m==0)
return 0;
vector<int> dp(m,0);
dp[0] = 1;
for(int i=0;i<n;i++)
for(int j=0;j<m;j++)
{
if(obstacleGrid[i][j] == 1)
dp[j] = 0;
else if(j != 0)
dp[j] += dp[j-1];
}
return dp[m-1];
}
};
点评:
与上题类似,空间复杂度可以优化成O(n);