面试题10- II. 青蛙跳台阶问题

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

答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。

思路:动态规划

注意:temp=a; a=b; b=temp+b;return b等价于a, b = b, a+b; return a

class Solution:
    def numWays(self, n: int) -> int:
        # f(0)=f(1)=1 f(n)=f(n-1)+f(n-2)
        a = 1
        b = 1
        for _ in range(2, n+1):
            temp = a
            a = b
            b = temp + b
        return b%1000000007     
发布了50 篇原创文章 · 获赞 14 · 访问量 2564

猜你喜欢

转载自blog.csdn.net/weixin_43455338/article/details/104836049
今日推荐