70.爬楼梯
解法:动态规划
转移函数:f(n) = f(n-1)+f(n-2)
解释:从第n阶楼梯倒退一步,因为一次要么爬一层,要么爬两层,所以只能从第 n-1 阶或者第 n-2 爬到第n阶,那么第n阶的爬法就是第n-1阶的爬法和第n-2阶爬法之和。
注意:
- 因为n=2前面只有一阶,所以我们定义一个n=0,f(0) = 1。
- 在实现的时候,可以使用递归或者滚动数组的方式,这里使用滚动数组,空间复杂度会更小。
代码:
class Solution:
def climbStairs(self, n: int) -> int:
#动态规划
ans = 1 #n>1,可以直接初始化返回值为f(1)=1
p = 1 #初始化指针,指向f(0) = 1
q = 1 #指向f(1) = 1
a = 2
while a <= n:
a += 1
ans = p + q
p = q #数组左移
q = ans
return ans
时间复杂度:O(n)
空间复杂度:O(1)