[LeetCode]70、爬楼梯

题目描述

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

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

注意：给定 n 是一个正整数。

示例 1：

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

1. 1 阶 + 1 阶
2. 2 阶
   示例 2：

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

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

思路解析

f(n) = f(n-1) + f(n-2)
斐波那契数列
优化可以用一个含有两个数字的数组
该优化思路就是先对res[0]更新

class Solution:
    def climbStairs(self, n: int) -> int:
        res = [1, 2]
        if n <= 2:
            return res[n-1]
        for i in range(3, n+1):
            res[(i+1)%2] = res[0] + res[1]
        return res[(n+1)%2]

AC70