70. 爬楼梯
假设你正在爬楼梯。需要 n 阶你才能到达楼顶。
每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?
注意:给定 n 是一个正整数。
示例 1:
输入: 2
输出: 2
解释: 有两种方法可以爬到楼顶。
- 1 阶 + 1 阶
- 2 阶
示例 2:
输入: 3
输出: 3
解释: 有三种方法可以爬到楼顶。
- 1 阶 + 1 阶 + 1 阶
- 1 阶 + 2 阶
- 2 阶 + 1 阶
第一种解法会超时,使用有备忘录的第二种解法。
class Solution:
def climbStairs(self, n: int) -> int:
if n == 1: return 1
if n == 2: return 2
return self.climbStairs(n-1)+self.climbStairs(n-2)
class Solution:
def climbStairs(self, n: int) -> int:
memo = {}#建立一个字典用于记录已经使用过的量
memo[1] = 1
memo[2] = 2
def climb(num):
if num in memo: return memo[num]
ans = climb(num-1) + climb(num-2)
memo[num] = ans
return ans
return climb(n)
if __name__ == "__main__":
s = Solution()
print(s.climbStairs(10))