【LeetCode】70. 爬楼梯（C++）

题目描述

假设你正在爬楼梯。需要 n 阶你才能到达楼顶。

每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢？

示例 1：

输入：n = 2
输出：2
解释：有两种方法可以爬到楼顶。

• 1. 1 阶 + 1 阶
• 2. 2 阶

示例 2：

输入：n = 3
输出：3
解释：有三种方法可以爬到楼顶。

• 1. 1 阶 + 1 阶 + 1 阶
• 2. 1 阶 + 2 阶
• 3. 2 阶 + 1 阶

提示：

• 1 <= n <= 45

---

思路：动态规划

1. dp[i] 的定义为：爬到第 i 层楼梯，有 dp[i] 种方法
2. 确定递推公式：dp[i] = dp[i - 1] + dp[i - 2];
3. 数组的初始化：题目提示 n >= 1，故无需考虑 dp[0] 的初始化。dp[1] = 1;  dp[2] = 2;
4. 遍历顺序：从前往后
5. 打印验证，结果正确

    // 时间复杂度：O(n)
    // 空间复杂度：O(n)
    class Solution {
    public:
        int climbStairs(int n) {
            if (n < 3) return n;
            int dp[n + 1];
            dp[1] = 1;
            dp[2] = 2;
            for (int i = 3; i <= n; i++) {
                dp[i] = dp[i - 1] + dp[i - 2];
            }
            return dp[n];
        }
    };

改进：优化空间复杂度

    // 时间复杂度：O(n)
    // 空间复杂度：O(1)
    class Solution {
    public:
        int climbStairs(int n) {
            if (n < 3) return n;
            int dp[3];
            dp[1] = 1;
            dp[2] = 2;
            for (int i = 3; i <= n; i++) {
                int sum = dp[1] + dp[2];
                dp[1] = dp[2];
                dp[2] = sum;
            }
            return dp[2];
        }
    };