70. 爬楼梯 leetcode

最新推荐文章于 2024-10-12 23:06:01 发布

「已注销」

最新推荐文章于 2024-10-12 23:06:01 发布

阅读量143

点赞数

文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/mossxzzom/article/details/127171830

版权

70. 爬楼梯

假设你正在爬楼梯。需要 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 阶

优化前

class Solution:
    def climbStairs(self, n: int) -> int:
        return f(n)

def f(n):
    if n == 1:
        return 1
    if n == 2:
        return 2
    if n == 3:
        return 3
    return f(n-1) + f(n-2)

优化后

class Solution:
    def climbStairs(self, n: int) -> int:
        return f(n)

c = {}

def f(n):
    if n == 1:
        return 1
    if n == 2:
        return 2
    if n == 3:
        return 3
    if n  in c:
        return c[n]
    c[n] = f(n-1) + f(n-2)
    return c[n]