写一个函数,输入 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
提示:
0 <= n <= 100
解题思路
动态规划:
原理: 以斐波那契数列性质 f(n + 1) = f(n) + f(n - 1)为转移方程。
从计算效率、空间复杂度上看,动态规划是本题的最佳解法。
动态规划解析:
状态定义: 设 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 个数字。
空间复杂度优化:
若新建长度为 n 的 dp 列表,则空间复杂度为 O(N) 。
由于 dp 列表第 i 项只与第 i−1 和第 i−2 项有关,因此只需要初始化三个整形变量 sum, a, b ,利用辅助变量 sum 使 a, b 两数字交替前进即可 (具体实现见代码) 。
节省了 dp 列表空间,因此空间复杂度降至 O(1) 。
class Solution {
public:
int fib(int n) {
//动态规划 dp[i]=dp[i-1]+dp[i-2]
//此处a相当于dp[0]、b相当于dp[1]
int a = 0, b = 1, sum;//sum相当于dp[i]
for(int i = 0; i < n; i++){//i最大为n-1
sum = (a + b) % 1000000007;//a + b相当于dp[i]+dp[i + 1]
a = b;//此刻a=dp[i]、最后一轮a相当于dp[n];而题目要求返回F(n),
//因此最后应该返回a
b = sum;//此刻b=dp[i+1]
}
return a;
}
}
复杂度分析:
时间复杂度 O(N): 计算 f(n)需循环 n 次,每轮循环内计算操作使用 O(1)。
空间复杂度 O(1) : 几个标志变量使用常数大小的额外空间。