题目
写一个函数,输入 n ,求斐波那契(Fibonacci)数列的第 n 项(即 F(N))。斐波那契数列的定义如下:
F(0) = 0, F(1) = 1
F(N) = F(N - 1) + F(N - 2), 其中 N > 1.
斐波那契数列由 0 和 1 开始,之后的斐波那契数就是由之前的两数相加而得出。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
示例1:
输入:n = 2
输出:1
递归法:
python版本
class Solution(object):
def fib(self,n):
a,b=0,1
for _ in range(n):
a,b=b,a+b
return a%1000000007
C++版本:
此时如果你这样写法就超时了
class Solution {
public:
int a=0;
int b=1;
int sum=0;
int fib(int n) {
if(n==1) return 1;
if(n==0) return 0;
else{
return (fib(n-1)+fib(n-2))% 1000000007;;
}
}
};
中间存在很多次计算,比如fib(3),他需要fib(2)和fib(1),fib(4)需要fib(3)和fib(2),因而在计算过程中,每一次fib(2),fib(3)都需要重复不断的计算,浪费了大量的时间,所以我们需要减少这样的重复计算,因而利用一个哈希表,把计算过的fib(n)的值存在哈希表中,这样就可以减少大量运算
class Solution{
public:
unordered_map<int,int> result;
int dfs(int n){
if(n==1) return 1;
else if(n==0) return 0;
else if(result.count(n)) return result[n]; //判断如果已经算过,直接返回,即如果第n个键值已经在哈希表出现,说明已经算过
int a=dfs(n-1)% 1000000007;
result[n-1]=a;
int b=dfs(n-2)% 1000000007;
result[n-2]=b;
result[n]=(a+b)% 1000000007;
return result[n];
}
int fib(int n){
return dfs(n);
}
}
或者
class Solution{
public:
int fib(int n){
vector<int>dp;
for(int i=0;i<=n;++i){
if(i==0) dp.push_back(0);
else if(i==1) dp.push_back(1);
else {
dp.push_back((dp[n-1]+dp[n-2])%1000000007);
}
}
return dp[n];
}
};