代码随想录算法训练营第34天 | 62.不同路径、63.不同路径Ⅱ、343.整数拆分、96.不同的二叉搜索树
文章目录
62.不同路径
解题思路
机器人从(0,0)的位置出发,到达(m-1,n-1)终点
- 确定dp数组及下标的含义:dp[i][j]:表示从(0,0)出发,到达(i,j)有 dp[i][j] 条不同的路径
- 确定递推公式:dp[i][j]=dp[i-1][j]+dp[i][j-1]
- 初始化dp数组:从(0,0)到(i,0)的路径只一条,所以dp[i][0]=1,dp[0][j]=1
- 确定遍历顺序:从左到右
- 举例推导dp数组
代码实现
class Solution {
public:
int uniquePaths(int m, int n) {
vector<vector<int>> dp(m, vector<int>(n, 0));
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];
}
}
return dp[m - 1][n - 1];
}
};
题目总结
按照正常的动规五部曲分析即可解决。
63.不同路径Ⅱ
解题思路
本题和上一题的区别在于有了障碍物
- 确定dp数组及下标的含义:dp[i][j]:表示从(0,0)出发,到达(i,j)有 dp[i][j] 条不同的路径
- 确定递推公式:dp[i][j]=dp[i-1][j]+dp[i][j-1],注意只有当(i,j)位置没有障碍物时,再推导
- 初始化dp数组:从(0,0)到(i,0)的路径只一条,所以dp[i][0]=1,dp[0][j]=1
- 确定遍历顺序:从左到右,注意碰到障碍物的情况
- 举例推导dp数组
代码实现
class Solution {
public:
int uniquePathsWithObstacles(vector<vector<int>>& obstacleGrid) {
int m = obstacleGrid.size();
int n = obstacleGrid[0].size();
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];
}
};
题目总结
遇到障碍物时,保持 dp[i][j] 为0即可。
343.整数拆分
解题思路
- 确定dp数组及下标的含义:dp[i]:拆分数字 i,可以得到的最大的乘积为 dp[i]
- 确定递推公式
从1开始遍历 j,有两种渠道可以得到 dp[i]:j×(i-j) 和 j×dp[i-j],j 是从1开始遍历的,在遍历的过程中已经计算了拆分 j 的情况。从1开始遍历,比较 j×(i-j) 和 j×dp[i-j] 后取最大的值,递推公式:dp[i]=max({dp[i], j×(i-j), j×dp[i-j]})
- 初始化dp数组:拆分1和0后的最大乘积没有意义,这里拆分数字2
- 确定遍历顺序:从前往后,先有 dp[i-j] 后有 dp[i]
- 举例推导dp数组
代码实现
class Solution {
public:
int integerBreak(int n) {
vector<int> dp(n + 1);
dp[2] = 1;
for (int i = 3; i <= n; i++) {
for (int j = 1; j < i - 1; j++) {
dp[i] = max({dp[i], j * (i - j), j * dp[i - j]});
}
}
return dp[n];
}
};
题目总结
本题还可以使用贪心来解决,将输入的 n 拆分成 k 个3,若剩下的数字是1,则和3拆分成2和2;若剩下的数字为2,那么直接相乘,这样可以保证乘积是最大的,这是基于数论的结论。
96.不同的二叉搜索树
解题思路
- 确定dp数组及下标的含义:dp[i]:组成的二叉搜索树的个数
- 确定递推公式
对于 i 个节点来说,依次从这些节点中挑选出一个节点作为 root,假设这个节点的位置是 j,那么 j 的取值范围是[1, i](闭区间)。其左子树就包含 j - 1 个节点,右子树就包含 i-j 个节点。
当固定了root节点时,二叉搜索树的数量=左子树的排列数量 * 右子树的排列数量。将每一个节点的情况累加,就是最终答案。
- 初始化dp数组:空树也是二叉搜索树,dp[0]=1
- 确定遍历顺序
- 举例推导dp数组
代码实现
class Solution {
public:
int numTrees(int n) {
vector<int> dp(n + 1);
dp[0] = 1;
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= i; j++) {
dp[i] += dp[j - 1] * dp[i - j];
}
}
return dp[n];
}
};
题目总结
本题的难点在于递推公式的确定,将大问题拆分为小问题解决,可以更好地分析出来。