假设你正在爬楼梯。需要 n 阶你才能到达楼顶。
每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?
注意:给定 n 是一个正整数。
示例 1:
输入: 2
输出: 2
解释: 有两种方法可以爬到楼顶。
1. 1 阶 + 1 阶
2. 2 阶
示例 2:
输入: 3
输出: 3
解释: 有三种方法可以爬到楼顶。
1. 1 阶 + 1 阶 + 1 阶
2. 1 阶 + 2 阶
3. 2 阶 + 1 阶
思路:
典型的动态规划问题,设输入为n,此时第一步有两种走法。
1.先走1个台阶,则还剩n-1个台阶。即转化为求输入为n-1的值。
2.先走两个台阶,还剩n-2个台阶,转化为求输入为n-2的值。
因此可以得到状态方程
d
p
[
1
]
=
1
,
d
p
[
2
]
=
2
dp[1]=1,dp[2]=2
dp[1]=1,dp[2]=2
d
p
[
n
]
=
d
p
[
n
−
1
]
+
d
p
[
n
−
2
]
,
(
n
>
2
)
dp[n]=dp[n-1]+dp[n-2] ,(n>2)
dp[n]=dp[n−1]+dp[n−2],(n>2)
其实就是个斐波那契数列。求斐波那契数列的方法就很多了,递归非递归什么的。不过递归的话时间复杂度
O
(
2
n
)
O(2^n)
O(2n)。
#o(n)
class Solution:
def climbStairs(self, n: int) -> int:
if n==1:
return 1
if n==2:
return 2
l=[0]*(n+1)
l[1]=1
l[2]=2
for i in range(3,n+1):
l[i]=l[i-1]+l[i-2]
return l[n]