https://leetcode.cn/problems/climbing-stairs/
假设你正在爬楼梯。需要 n 阶你才能到达楼顶。每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?
示例 1:
输入:n = 2
输出:2
解释:有两种方法可以爬到楼顶。
1 阶 + 1 阶
2 阶
class Solution:
def climbStairs(self, n: int) -> int:
# 难度:☆☆☆
# 方法1:带memo的递归(普通递归会超时,不另外设置函数fn的方法也会超时)
# 1个楼梯(1)
# 2个楼梯(1,1)(2)
# 3个楼梯(1,2)(2,1)(1,1,1)
# 4个楼梯(2,2)(2,1,1)(1,2,1)(1,1,2)(1,1,1,1)
# 5个楼梯(2,2,1)(2,1,2)(1,2,2)(2,1,1,1)(1,2,1,1)(1,1,2,1)(1,1,1,2)(1,1,1,1,1)
# n = 1 1
# n = 2 2
# n = 3 3 (n-1) + (n-2)
# n = 4 5
# n = 6 8
# memo = {}
# def f(n):
# if n <= 2:
# return n
# elif n in memo:
# return memo[n]
# else:
# memo[n] = f(n - 1) + f(n - 2)
# return memo[n]
# return f(n)
# 方法2:动态规划
dp = [0] * (n + 1) # ① 定义dp:dp[i] 到达i有多少种方法
dp[0], dp[1] = 1, 1 # ② 初始化
# ③ 遍历递归(状态转移方程)
for i in range(2, n + 1):
dp[i] = dp[i - 1] + dp[i - 2]
return dp[-1]