70. 爬楼梯

题目

70. 爬楼梯（简单）

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

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

示例 1：

输入：

2

输出：

2

解释：有两种方法可以爬到楼顶。

1. 1 阶 + 1 阶
2. 2 阶

示例 2：

输入：

3

输出：

3

解释：有三种方法可以爬到楼顶。

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

提示：

• 1 <= n <= 45

思路

斐波那契递推：到达第 $n$ 阶的最后一步要么从 $n-1$ 阶迈 1 步、要么从 $n-2$ 阶迈 2 步，故 $f(n) = f(n-1) + f(n-2)$。用两个变量滚动即可，空间 $O(1)$。

代码

class Solution:
    def climbStairs(self, n: int) -> int:
        a, b = 1, 1
        for _ in range(n - 1):
            a, b = b, a + b
        return b