首页 > 代码库 > Unique Paths II

Unique Paths II

  这题在Unique Paths的基础上增加了一些obstacle的位置,应该说增加的难度不大,但是写的时候对细节的要求多了很多,比如,第一列的初始化会受到之前行的第一列的结果的制约。另外对第一行的初始化,也要分if else赋值。很容易出现初始化不正确的情况。

  代码:

  

class Solution {public:    int uniquePathsWithObstacles(vector<vector<int> > &obstacleGrid) {        if(obstacleGrid[0][0]==1)            return 0;        int m=obstacleGrid.size();        int n=obstacleGrid[0].size();        int * row1=new int[n];        int * row2=new int[n];        row1[0]=obstacleGrid[0][0]==1?0:1;        int obs=0;//indicate the first row is obscaled        for(int i=1;i<n;i++)        {            if(row1[i-1]==0||obstacleGrid[0][i]==1)                row1[i]=0;            else                row1[i]=1;        }        for(int i=1;i<m;i++)        {            row2[0]=obstacleGrid[i][0]==1?0:1;            if(obstacleGrid[i][0]==1||obs==1)            {                obs=1;                row2[0]=0;            }            for(int j=1;j<n;j++)            {                if(obstacleGrid[i][j]==1)                    row2[j]=0;                else                {                    row2[j]=row2[j-1]+row1[j];                }            }            row1=row2;        }        return row1[n-1];    }};