题目描述:
一个机器人位于一个 m x n 网格的左上角 (起始点在下图中标记为 “Start” )。
机器人每次只能向下或者向右移动一步。机器人试图达到网格的右下角
现在考虑网格中有障碍物。那么从左上角到右下角将会有多少条不同的路径?
网格中的障碍物和空位置分别用 1 和 0 来表示。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/unique-paths-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
典型的动态规划问题,dp[i][j]的含义表示走到这个网格的不同路径的个数。显然在有障碍物的格子上这个值为0。在其他情况下dp[i][j]=dp[i-1][j]+dp[i[[j-1]初始化第一行和第一列以后就可以开始遍历dp数组了。注意初始化行和列的时候一旦遇到障碍物那么障碍物后面的格子都要初始化为0。
代码:
class Solution {
public:
int uniquePathsWithObstacles(vector<vector<int>>& obstacleGrid) {
//获取行数和列数
int m = obstacleGrid.size(),n=obstacleGrid[0].size();int dp[m][n];
//判断一行一列的特殊情况
if(m==1){
for(int i=0;i<n;i++)if(obstacleGrid[0][i]==1)return 0;
return 1;
}
if(n==1){
for(int i=0;i<m;i++)if(obstacleGrid[i][0]==1)return 0;
return 1;
}
//把dp数组全部初始化成0
for(int i=0;i<m;i++){
for(int j=0;j<n;j++){
dp[i][j]=0;
}
}
//根据障碍物重新初始化dp数组的第一行和第一列
for(int i=0;i<m;i++){
if(obstacleGrid[i][0]==0){
dp[i][0]=1;
}else if(obstacleGrid[i][0]==1){
dp[i][0]=0;break;
}
}
for(int i=0;i<n;i++){
if(obstacleGrid[0][i]==0){
dp[0][i]=1;
}else if(obstacleGrid[0][i]==1){
dp[0][i]=0;break;
}
}
//遍历没有障碍物的格子
for(int i=0;i<m;i++){
for(int j=0;j<n;j++){
cout<<dp[i][j]<<' ';
}
cout<<endl;
}
for(int i=1;i<m;i++){
for(int j=1;j<n;j++){
if(obstacleGrid[i][j]==0){
dp[i][j]+=dp[i-1][j]+dp[i][j-1];
}
}
}
//可以查看dp数组的最终结果
for(int i=0;i<m;i++){
for(int j=0;j<n;j++){
cout<<dp[i][j]<<' ';
}
cout<<endl;
}
return dp[m-1][n-1];
}
};
感悟:
在不知道哪里出错的情况下可以输出dp数组看看是不是我想要的结果。