Leetcode 第70题(爬楼梯)

70.爬楼梯
Leetcode 第70题(爬楼梯)_第1张图片
解法:动态规划

转移函数:f(n) = f(n-1)+f(n-2)

解释:从第n阶楼梯倒退一步,因为一次要么爬一层,要么爬两层,所以只能从第 n-1 阶或者第 n-2 爬到第n阶,那么第n阶的爬法就是第n-1阶的爬法和第n-2阶爬法之和。

注意:

  1. 因为n=2前面只有一阶,所以我们定义一个n=0,f(0) = 1。
  2. 在实现的时候,可以使用递归或者滚动数组的方式,这里使用滚动数组,空间复杂度会更小。

代码:

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)

你可能感兴趣的:(Leetcode,python,动态规划)