题目描述:
写一个函数,输入 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。
示例 1:
输入:n = 2
输出:1
示例 2:
输入:n = 5
输出:5
链接:https://leetcode-cn.com/problems/fei-bo-na-qi-shu-lie-lcof
斐波那契数列即:fib(n) = fib(n - 1) + fib(n - 2)
从该图中可发现,斐波那契数列中,存在着许多重复计算,即许多重叠的子问题。常规思路求解时,可采用递归,但当递归深度太大时,就会出现栈溢出,超时等现象。所以考虑用动态规划来解决该问题。
1.递归
注:该问题可用递归解决,但递归层数太多,会超时,一般不采取这种方法。
代码:
class Solution {
public:
int _fib(int n)
{
if (n == 0)
return 0;
if (n <= 2)
return 1;
return _fib(n - 1) + _fib(n - 2);
}
int fib(int n) {
return _fib(n) % 1000000007;
}
};
2.动态规划
思路:动态规划转移方程为:dp[i] = dp[i - 1] + dp[i - 2]
。借助vector开辟空间使用即可。
代码:
时间复杂度O(N)、空间复杂度O(N)
class Solution {
public:
int fib(int n) {
if(n == 0)
return 0;
if(n <= 2)
return 1;
vector<int> dp(n + 1, 0);
dp[1] = dp[2] = 1;
for(int i = 3; i <= n; ++i)
dp[i] = (dp[i - 1] + dp[i - 2]) % 1000000007;
return dp[n];
}
};
3.状态压缩
思路:上述动态规划中,发现始终只是改变3个值。则不必申请n个空间。进一步降低空间复杂度。
代码:
时间复杂度O(N)、空间复杂度O(1)
class Solution {
public:
int fib(int n) {
if(n == 0)
return 0;
if(n <= 2)
return 1;
int first = 1;
int second = 1;
int third;
for(int i = 3; i <= n; ++i)
{
third = (first + second) % 1000000007;
first = second;
second = third;
}
return third;
}
};