写一个函数,输入 n ,求斐波那契(Fibonacci)数列的第 n 项。斐波那契数列的定义如下:
F(0) = 0, F(1) = 1
F(N) = F(N - 1) + F(N - 2), 其中 1 < N <= 100.
斐波那契数列由 0 和 1 开始,之后的斐波那契数就是由之前的两数相加而得出。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
后一项 = 前一项+前前一项 首先我们会想到使用递归:
public int fib(int n) {
return n < 2 ? n : (fib(n - 1) + fib(n - 2)) % (int)(1e9 + 7);
}
递归代码非常简洁,但是递归的缺点是有时间和空间的消耗,每次函数调用都需要在内存栈中分配空间以保存参数、返回地址及临时变量,且往栈里压入数据和弹出数据都需要时间。因此递归的效率不如循环。本题使用递归代码,当n大于一定值时(如43)会超出时间限制,该算法时间复杂度为O(n^2), 空间复杂度为O(1)
优化代码,我们想到,获取f(x)、f(x-1)、f(x-2)等非常方便的数据结构是数组,我们可以把所有斐波那契数列放在一个数组中,使用arr(n)即可求出第n项数。
public int fib(int n) {
int[] arr = new int[n + 1];
for (int i = 0; i < n + 1; i++) {
if (i > 1)
arr[i] = (arr[i - 1] + arr[i - 2]) % 1000000007;
else
arr[i] = i;
}
return arr[n];
}
一遍循环 时间复杂度:O(n) 空间复杂度: O(n)
继续优化,降低空间复杂度,使用动态规划
public int fib(int n) {
int[] res = {0, 1};
if (n < 2) return res[n];
int one = 1;
int two = 0;
int fibN = 0;
for (int i = 2; i <= n; i++) {
fibN = (one + two) % 1000000007;
two = one;
one = fibN;
}
return fibN;
}
时间复杂度:O(n)
空间复杂度:O(1)