Leetcode Day3 2

一只青蛙一次可以跳上1级台阶，也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。

答案需要取模 1e9+7（1000000007），如计算初始结果为：1000000008，请返回 1。

太菜了我，直接拿上一题的来写，想使用下range结果大失败T-T

class Solution:

    def numWays(self, n: int) -> int:

        def solve(n):

            res = 0

            tmp1 = 2

            tmp = [1, 1]

            if n == 1:

                return 1

            elif n == 0:

                return 1

            else:

                while tmp1<=n:

                    res = int((tmp[tmp1 - 1] + tmp[tmp1 - 2]) % (1e9 + 7))

                    tmp.append(res)

                    tmp1+=1

            return res

        return solve(n)

对着前一题的最佳解改了一下

啊啊一定要记住是return a啊，不是return res，不然就多算了一次了。

不过不知道为什么我的速度比佬的答案要快一些，估计每次取余后面的数就小了，嗯。