题目
假设你正在爬楼梯。需要 n (正整数) 阶你才能到达楼顶。每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?
示例
input: 2
output: 2
解释:有两种方法可以爬到楼顶。
1. 1 阶 + 1 阶
2. 2 阶
input: 3
output: 3
解释:有三种方法可以爬到楼顶。
1. 1 阶 + 1 阶 + 1 阶
2. 1 阶 + 2 阶
3. 2 阶 + 1 阶
解法
动态规划 ( Python )
class Solution:
def climbStairs(self, n: int) -> int:
sub = [1,2]
for i in range(2, n):
sub.append(sub[i - 1] + sub[i - 2])
return sub[n - 1] if n != 0 else 1
基本思路
对于动态规划而言,关键是要找到原问题和子问题之间的关系。要爬上第n
阶楼梯,只有两种方法,一种是从第n - 1
阶爬一步,另一种是从第n - 2
阶爬两步,基于此,我们可以构造状态转移方程f(n) = f(n - 1) + f(n - 2)
。
复杂度分析
for
循环中需要计算从 2 到 n 所有子问题的爬楼梯方法,时间复杂度为
O
(
N
)
O(N)
O(N)。动态规划子问题的计算结果通常都需要用一个数组来保存,在这里也就是指sub
,长度为N,所以空间复杂度为
O
(
N
)
O(N)
O(N) 。