leetcode---70.爬楼梯

题目描述

假设你正在爬楼梯。需要 n 阶你才能到达楼顶。

每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢？

**注意：**给定 n 是一个正整数。

示例

输入： 3
输出： 3
解释： 有三种方法可以爬到楼顶。
1.  1 阶 + 1 阶 + 1 阶
2.  1 阶 + 2 阶
3.  2 阶 + 1 阶

解题思路

当 $n=1$ 时，只有一种方法，当 $2\le n$时，考虑动态规划，建立一个 $n$ 维数组 $dp$，$dp[0]=1$，$dp[1]=2$，当 $i>1$，有状态转移方程 $dp[i]=dp[i-1]+dp[i-2]$。

代码

class Solution:
    def climbStairs(self, n: int) -> int:
        dp = [0] * n
        if n <= 1:
            return 1

        dp[0] = 1
        dp[1] = 2
        for i in range(2,n):
            dp[i] = dp[i-1] + dp[i-2]
        return dp[-1]

优化后代码

class Solution:
    def climbStairs(self, n: int) -> int:
        
        a = b = 1
        for i in range(2, n + 1):
            a, b = b, a + b
        return b

复杂度

• 时间复杂度 $O(n)$；
• 空间复杂度，优化前 $O(n)$，优化后 $O(1)$。