剑指-面试题10-1 斐波拉契数列 10-2青蛙跳台阶

斐波拉契数列、青蛙跳台阶

题目1
写一个函数，输入 n ，求斐波那契（Fibonacci）数列的第 n 项。斐波那契数列的定义如下：
F(0) = 0, F(1) = 1
F(N) = F(N - 1) + F(N - 2), 其中 N > 1.
斐波那契数列由 0 和 1 开始，之后的斐波那契数就是由之前的两数相加而得出。
答案需要取模 1e9+7（1000000007），如计算初始结果为：1000000008，请返回 1。

思路：
递归时间太多，用循环，且C++用long long；
为了匹配leetcode的测试用例，多了一步n++；
运行中取模而不是对结果取模。

C++

class Solution {
public:
    int fib(int n) {
        int result[2]={0,1};                //n=0,1时的情况
        if (n<2)                    
            return result[n];        
        long long fib1=0;
        long long fib2=1;
        long long fibN=0;
        n++;
        for(int i=2; i<=n; ++i)
        {
            fibN= (fib1+fib2) % 1000000007;
            fib2 = fib1;
            fib1 = fibN;
        }
        return fibN;
    }
};

python

class Solution:
    def fib(self, n: int) -> int:
        if n<2:
            return n
        fib1 = 0
        fib2 = 1
        fibn = 0
        n+=1
        for i in range(2,n+1):
            fibn = (fib1+fib2) % 1000000007
            fib2 = fib1
            fib1 = fibn
        return fibn

题目2
一只青蛙一次可以跳上1级台阶，也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。
答案需要取模 1e9+7（1000000007），如计算初始结果为：1000000008，请返回 1。

思路：

C++

class Solution {
public:
    int numWays(int n) {
        if (n<2)                    
            return 1;        
        long long fib1=0;
        long long fib2=1;
        long long fibN=0;
        n += 2;   // 为了leetcode通过
        for(int i=2; i<=n; ++i)
        {
            fibN= (fib1+fib2) % 1000000007;
            fib2 = fib1;
            fib1 = fibN;
        }
        return fibN;
    }
};

python

class Solution:
    def numWays(self, n: int) -> int:
        if n<2:
            return 1
        fib1 = 0
        fib2 = 1
        fibn = 0
        n+=2
        for i in range(2,n+1):
            fibn = (fib1+fib2) % 1000000007
            fib2 = fib1
            fib1 = fibn
        return fibn