第四题:
力扣(LeetCode)官网 - 全球极客挚爱的技术成长平台
要注意矩阵的第一行和第一列初始化!
class Solution {
public:
int uniquePaths(int m, int n) {
if (m < 2 || n < 2) return 1;
vector<vector<int>> dp(m, vector<int> (n));
//dp初始化
for (int i = 0; i < m; i++) {
dp[i][0] = 1;
}
for (int j = 0; j < n; j++) {
dp[0][j] = 1;
}
for (int i = 1; i < m; i++) {
for (int j = 1; j < n; j++) {
//递推公式
dp[i][j] = dp[i-1][j] + dp[i][j-1];
}
}
int res = dp[m-1][n-1];
return res;
}
};
第五题:
力扣(LeetCode)官网 - 全球极客挚爱的技术成长平台
这道题和前一道的区别在于,dp初始化和递推公式的使用条件:
- dp初始化:当obs位于第一行或者第一列时,obs所占格(包括它自己)往右或者往下的dp应该初始化为0,因为在第一行和第一列只能向右走或者向下走,一旦被obs挡住后面的格子都走不了了;
- 递推公式:当dp[i][j]不是障碍物所在格子时,才和上一题一样,使用递推公式求解dp,如果obs[i][j]为1,表示障碍物就在当前这个格子,那么这个格子的dp是无解的,可以直接跳过了;
- 此外障碍物在起点和终点的时候,也是直接无解的。
class Solution {
public:
int uniquePathsWithObstacles(vector<vector<int>>& obstacleGrid) {
int n = obstacleGrid[0].size();
int m = obstacleGrid.size();
if (obstacleGrid[m - 1][n - 1] == 1 || obstacleGrid[0][0] == 1) //如果在起点或终点出现了障碍,直接返回0
return 0;
vector<vector<int>> dp(m, vector<int> (n, 0));
for (int i = 0; i < m; i++) {
if (obstacleGrid[i][0] == 1) {
break;
}
dp[i][0] = 1;
}
for (int j = 0; j < n; j++) {
if (obstacleGrid[0][j] == 1) {
break;
}
dp[0][j] = 1;
}
for (int i = 1; i < m; i++) {
for (int j = 1; j < n; j++) {
if (obstacleGrid[i][j] == 1) continue;
dp[i][j] = dp[i-1][j] + dp[i][j-1];
}
}
return dp[m-1][n-1];
}
};
代码随想录的代码:
class Solution {
public:
int uniquePathsWithObstacles(vector<vector<int>>& obstacleGrid) {
int m = obstacleGrid.size();
int n = obstacleGrid[0].size();
if (obstacleGrid[m - 1][n - 1] == 1 || obstacleGrid[0][0] == 1) //如果在起点或终点出现了障碍,直接返回0
return 0;
vector<vector<int>> dp(m, vector<int>(n, 0));
for (int i = 0; i < m && obstacleGrid[i][0] == 0; i++) dp[i][0] = 1;
for (int j = 0; j < n && obstacleGrid[0][j] == 0; j++) dp[0][j] = 1;
for (int i = 1; i < m; i++) {
for (int j = 1; j < n; j++) {
if (obstacleGrid[i][j] == 1) continue;
dp[i][j] = dp[i - 1][j] + dp[i][j - 1];
}
}
return dp[m - 1][n - 1];
}
};