爬楼梯
题目描述
假设你正在爬楼梯。需要 n 阶你才能到达楼顶。
每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?
注意:给定 n 是一个正整数。
示例 1:
输入: 2
输出: 2
解释: 有两种方法可以爬到楼顶。
- 1 阶 + 1 阶
- 2 阶
示例 2:
输入: 3
输出: 3
解释: 有三种方法可以爬到楼顶。
- 1 阶 + 1 阶 + 1 阶
- 1 阶 + 2 阶
- 2 阶 + 1 阶
思路
思路:
f[0]=0
f[1]=1
n=2时:可以1+1或者 2,因此有两种
n=3时:横着放还剩 f[2],竖着放还剩f[1]
因此 f[n]=f[n-1]+f[n-2]
代码
非递归方法
##### 非递归方法
class Solution(object):
def climbStairs(self, n):
"""
:type n: int
:rtype: int
"""
if n == 0:
return 0
f_1 = 0
f_2 = 1
for i in range(1, n + 1):
f_n = f_1 + f_2
f_1 = f_2
f_2 = f_n
return f_n
递归方式,会超时
class Solution(object):
def climbStairs(self, n):
"""
:type n: int
:rtype: int
"""
if n == 0:
return 0
if n == 1:
return 1
if n == 2:
return 2
f_n = self.climbStairs(n - 1) + self.climbStairs(n - 2)
return f_n