算法题1上台阶问题

Posted 2020-10-30 MKYAN

题目：有一楼梯共m级，刚开始时你在第一级，若每次只能跨上一级或者二级，要走上m级，共有多少走法？注：规定从一级到一级有0种走法。

给定一个正整数int n，请返回一个数，代表上楼的方式数。保证n小于等于100。为了防止溢出，请返回结果Mod 1000000007的值。

测试样例：

3

返回：2

 

来源：牛客网：京东2016算法工程师笔试题

 1 class goUpstairs:
 2     def countWays(self,n):
 3         if n==1:
 4             return 0
 5         if n==2:
 6             return 1
 7         if n==3:
 8             return 2
 9         else:
10             return (self.countWays(n-1)+self.countWays(n-2)) % 1000000007  # 为了防止溢出
11 
12 
13 class goUpstairs2:
14     def countWays(self,n):
15         res=[1,1]
16         if n==1:
17             return 0
18         while len(res)<n:
19             res.append(res[-1]+res[-2])
20         return res[-1] % 1000000007
21 
22 if __name__=="__main__":
23     # for i in range(1,10):
24     #     print(goUpstairs2().countWays(i))
25     for i in range(1, 10):
26         print(goUpstairs().countWays(i))

本题用递归的斐波那契数列算法会超时，所以可以用方法二。