题目描述:一只青蛙一次可以跳上1级台阶,也可以跳上2级。求该青蛙跳上一个n级的台阶总共有多少种跳法。
思路:青蛙跳台阶是和斐波那契数列一样典型的问题,可参考我上一篇博文,这也是一个典型的当前值只和上两个值有关的问题,step(n=1)=1 ; step(n=2)=2 ; step(n=3) =step(n=1)+step(n=2),所以同样的,只要保留上两个的值就好啦,offer就拿到啦(并没有)。
代码:
class Solution:
def jumpFloor(self, number):
if number <0:
return -1
elif number == 0:
return 0
elif number == 1:
return 1
elif number == 2:
return 2
else:
n1,n2 = 1,2
step = 0
for i in range(3,number+1):
step = n1+n2
n1 = n2
n2 = step
return step