假设你正在爬楼梯。需要 n 阶你才能到达楼顶。
每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?
示例 1:
输入:n = 2
输出:2
***解释:***有两种方法可以爬到楼顶。
- 1 阶 + 1 阶
- 2 阶
示例 2:
输入:n = 3
输出:3
***解释:***有三种方法可以爬到楼顶。
- 1 阶 + 1 阶 + 1 阶
- 1 阶 + 2 阶
- 2 阶 + 1 阶
提示:
1 <= n <= 45
思路:
(1)递归
(2)动态规划:
记忆化更新dp数组即可
Python代码:
class Solution(object):
def climbStairs(self, n):
"""
:type n: int
:rtype: int
"""
dp = [int(0)]*(n+1) #dp
if n==1:
return 1
if n==2:
return 2
dp[1]=1
dp[2]=2
for i in range(3,n+1):
dp[i]=dp[i-1]+dp[i-2]
return dp[n]
(3)滚动数组
第二种方法将走到每一层楼梯的方法数都保存进一个数组,但是求走到第n层楼的方法数,只需要保存n-1和n-2层的方法数即可。用滚动数组法,每次只保存所求层数的前两层的结果即可。在法2的基础上节省了空间。
python代码:
class Solution(object):
def climbStairs(self, n):
"""
:type n: int
:rtype: int
"""
if n==1:
return 1
if n==2:
return 2
first = 1
second = 2
for i in range(3,n+1):
third = first+second
first=second
second=third
return third