题目描述
写一个函数,输入 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
示例 2:
输入:n = 5 输出:5
解题步骤
此题如果使用递归会超出时间限制
使用动态规划可以解决。
动态规划解析:
状态定义: 设 dp为一维数组,其中 dp[i]的值代表 斐波那契数列第 i个数字 。
转移方程: dp[i + 1] = dp[i] + dp[i - 1] ,即对应数列定义 f(n + 1) = f(n) + f(n - 1) ;
初始状态: dp[0] = 0, dp[1] = 1,即初始化前两个数字;
返回值: dp[n] ,即斐波那契数列的第 n 个数字。
由于 dp 列表第 i 项只与第 i-1 和第 i-2 项有关,因此只需要初始化三个整形变量 sum, a, b,利用辅助变量 sum使 a, b 两数字交替前进即可 。
由于节省了 dp 列表空间,因此空间复杂度降至 O(1)。
代码如下
class Solution {
public:
int fib(int n) {
if(n==0)
{
return 0;
}
if(n==1)
{
return 1;
}
int a=0;
int b=1;
int sum=0;
for(int i=2; i<=n; ++i){
sum=(a+b)%1000000007;
a=b;
b=sum;
}
return sum;
}
};