目录

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;
    }
};