LeetCode 63. Unique Paths II C++

Problem

题目链接

Solution

此题为LeetCode62的后续，在62题的基础上，增加了障碍限制（若格子上有障碍，则无法通过这个格子）

有了62题的基础，我们可以直接写一维dp

若地图中某个格子有障碍，则该格子不能通行，该格子方案数都需要置0。此处有个特别注意的地方，若第一行中有障碍，那么不光该格子需要置0，且第一行中这个位置后所有格子都需要置0（因为只能向下或向右移动，不可能绕过这个障碍再回到第一行了）。

若地图中某格子没有障碍，那么该格子状态转移方程和62题是一样的， dp[j]+=dp[j-1];，直接套用就可以。

边界情况：第一行方案数都为1（还要按上面所说检查一下第一行有没有障碍）。

编写代码：

class Solution {
public:
    int uniquePathsWithObstacles(vector<vector<int>>& obstacleGrid) {
        int m=obstacleGrid.size();
        int n=obstacleGrid[0].size();
        
        int i;
        for(i=0;i<n;++i) 
            if(obstacleGrid[0][i]==1) break;
        
        long long dp[n];
        for(int j=0;j<i;++j) dp[j]=1;
        for(;i<n;++i) dp[i]=0;
        
        for(i=1;i<m;++i)
        {
            for(int j=0;j<n;++j)
            {
                if(obstacleGrid[i][j]==1)
                {
                    dp[j]=0;
                }
                else if(j>0) //第一个格子只能是从上面的格子往下走一步得来的，因此无需考虑j==0
                {
                    dp[j]+=dp[j-1];
                }
            }
        }
        return dp[n-1];
    }
};

Result