目录
1 70. 爬楼梯
1.1 基本思路
1.2 官方题解
2 118. 杨辉三角
3 198. 打家劫舍
菜鸟做题,语言是 C++
1 70. 爬楼梯
核心思想:把总问题拆解为若干子问题。
- 总问题:上到 5 楼的方式有多少种
- 子问题:上到 4 楼的方式有多少种、上到 3 楼的方式有多少种
- 总问题 = 子问题 1 + 子问题 2
因为题目要求每次只能爬 1 或 2 个台阶,所以子问题只有两个。
1.1 基本思路
假设按照英国算法要爬 6 层楼,如下图所示:
使用一个名为 dp 的数组来存储爬到每一层楼的方式种数。
由于我们只能从 4 或 3 楼爬到 5 楼,因此爬到 5 楼的方式种数 = 爬到 4 楼的方式种数 + 爬到 3 楼的方式种数,即 dp[5] = dp[4] + dp[3],以此类推。
特别地,对于 0 楼和 1 楼,由于只有一种方式爬到它们,因此直接设为 1 即可。
很不幸,上述方法需要维护一个长为 n 的数组 dp,因此空间复杂度是 O(n),而这样会超出内存限制,下面来看官方题解的奇妙方法。
1.2 官方题解
由 1.1 节的分析可知,dp[5] 只与 dp[4] 和 dp[3] 有关。因此在第 5 时刻,我们只需要记住 dp[4] 和 dp[3] 即可。换句话说,就是通过忘记其他 dp 元素来节省内存空间。
如下图所示。我们设置变量 p、q、r 来分别存储 dp[i - 2]、dp[i - 1]、dp[i],并不断更新这些变量。
可以把 p、q、r 理解为一个滑动窗口,每次同时向后移动一格,以此忘记不再需要的 dp 元素。
完整代码如下:
class Solution {
public:int climbStairs(int n) {int p = 0, q = 0, r = 1;for (int i = 0; i < n; ++i) {p = q;q = r;r = p + q;}return r;}
};
这个启动方式还是挺妙的:
int p = 0, q = 0, r = 1;
2 118. 杨辉三角
核心思想:把总问题拆解为若干子问题。
- 总问题:第 i 行第 j 列元素的值
- 子问题:第 i - 1 行第 j 列元素的值、第 i - 1 行第 j + 1 列元素的值
- 总问题 = 子问题 1 + 子问题 2
模仿上述动图初始化 ans 数组:
vector<vector<int>> ans(numRows);
for (int i = 1; i <= numRows; ++i) {ans[i - 1].resize(i);ans[i - 1][0] = 1;ans[i - 1][i - 1] = 1;
}
就是把那一圈 “1” 先填了,虽然笨,但貌似不影响效率。
完整代码如下:
class Solution {
public:vector<vector<int>> generate(int numRows) {vector<vector<int>> ans(numRows);for (int i = 1; i <= numRows; ++i) {ans[i - 1].resize(i);ans[i - 1][0] = 1;ans[i - 1][i - 1] = 1;}for (int i = 2; i < numRows; ++i) {for (int j = 1; j < ans[i].size() - 1; ++j) {ans[i][j] = ans[i - 1][j - 1] + ans[i - 1][j];}}return ans;}
};
3 198. 打家劫舍
核心思想:把总问题拆解为若干子问题。
- 总问题:打劫完第 i 家所获最大金额
- 子问题:打劫完第 i - 2 家所获最大金额、打劫完第 i - 3 家所获最大金额
- 总问题 = 子问题 1 + 子问题 2
Q:为什么只考虑打劫第 i - 2 家和第 i - 3 家?
A:假设 i = 5,如下图所示。对于第 i - 1 家,由于不能打劫相邻的两家,因此要想打劫第 i 家,就不能打劫第 i - 1 家。可以打劫第 i - 2 家或者第 i - 3 家,如情况一和情况二所示。对于第 i - 4 家,由于要使金额最大,因此肯定还会打劫第 i - 2 家,从而认为情况一等价于情况三。
综上,我们只需要考虑情况一和情况二即可。
同 70. 爬楼梯 的简化思路相同,我们同样可以只设置几个变量来存储历史打劫金额,而非使用一个 dp 数组。其中,h3 存储第 i - 3 家,h2 存储第 i - 2 家,h1 存储第 i - 1 家,h0 存储第 i 家。
完整代码如下:
class Solution {
public:int rob(vector<int>& nums) {int h3 = 0, h2 = 0, h1 = 0;int ans = INT_MIN;for (int i = 0; i < nums.size(); ++i) {int h0 = max(nums[i] + h3, nums[i] + h2);ans = max(ans, h0);h3 = h2;h2 = h1;h1 = h0;}return ans;}
};