一次面试中的算法题记录：爬楼梯

 

题目描述：
假设你正在爬楼梯。需要n阶你才能到达楼顶。
每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢？
注意：给定 n 是一个正整数，0 < n <= 100
示例 1：
输入： 2
输出： 2
解释： 有两种方法可以爬到楼顶。
1.  1 阶 + 1 阶
2.  2 阶
示例 2：
输入： 3
输出： 3
解释： 有三种方法可以爬到楼顶。
1.  1 阶 + 1 阶 + 1 阶
2.  1 阶 + 2 阶
3.  2 阶 + 1 阶
爬楼梯，很经典的题目，其实就是斐波那契数列，一开始我立马想到用递归，写出来后想了想复杂度太大了，也跟面试官说了这个的缺点，然后面试官说有没有更优化的方法，于是做了下修改

# 第一次的方法

def func1(n: int):
	if n == 1:
		return 1
	if n == 2:
		return 2
	if n > 2:
		return func(n - 1) + func(n - 2)



# print(func1(50))
# 需要很长时间

# 第二次的方法

def func2(n: int):
	if n == 1:
		return 1
	if n == 2:
		return 2
	if n > 2:
		a = 1
		b = 2
		while n > 2:
			a, b = b, a + b
			n -= 1
		return b

# print(func2(50))
# 结果：20365011074