题目要求
原题链接:
写一个函数,输入 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
解题过程
未优化的递归 | 时间复杂度: O ( 2 n ) O(2^n) O(2n) | 空间复杂度: O ( n ) O(n) O(n) | 不推荐
指数级别的时间复杂度,显然是不行的,所以需要优化!时间复杂度参考:斐波那契数列的三种算法以及复杂度
class Solution {
public int fib(int n) {
if(n < 2) return n;
return (fib(n - 1) + fib(n - 2));
}
}
优化后的递归(记忆化递归/搜索) | 时间复杂度: O ( n ) O(n) O(n) | 空间复杂度: O ( n ) O(n) O(n) | 推荐
根据递归的顺序,可以将计算过的值保存,方便下次递归调用的时候检查,如果该值计算过,则直接剪枝返回。
class Solution {
int mod = 1000000007;
int first = 0;
int second = 0;
int res = 0;
public int fib(int n) {
return recur(n, new HashMap<Integer, Integer>());
}
public int recur(int n, Map<Integer, Integer> map){
if(n < 2) return n;
// map中存在n,说明这个值已经计算过了
if(map.containsKey(n)) return map.get(n);
// 计算并且添加 n - 1节点的值
first = recur(n - 1, map) % mod;
map.put(n - 1, first);
// 计算并且添加 n - 2节点的值
second = recur(n - 2, map) % mod;
map.put(n - 2, second);
// 计算并且添加 n 节点的值
res = (first + second) % mod;
map.put(n, res);
// 返回结果
return res;
}
}
除了用Map,也可以用数组来存存储计算过的值 | 时间复杂度: O ( n ) O(n) O(n) | 空间复杂度: O ( n ) O(n) O(n)
class Solution {
int mod = 1000000007;
int first = 0;
int second = 0;
int res = 0;
int[] arr = new int[101];
public int fib(int n){
if(n < 2) return n;
if(arr[n] != 0) return arr[n];
first = fib(n - 1) % mod;
arr[n - 1] = first;
second = fib(n - 2) % mod;
arr[n - 2] = second;
res = (first + second) % mod;
arr[n] = res;
return res;
}
}
优化后的递归(尾递归) | 时间复杂度: O ( n ) O(n) O(n) | 空间复杂度: O ( 1 ) O(1) O(1) | 推荐
为什么尾递归的方式要比传统递归的方式时间复杂度更低呢?
不同于传统的递归方式,尾递归在调用方法的时候,就将数据作为形参进行了缓存累加,作为下一次调用的输入,大大节省了开销。而传统的递归产生的栈的层次像二叉树一样,以指数级增长,但是现在栈的层次却像是数组,变成线性增长了,有点dp的味道。
class Solution {
int mod = 1000000007;
public int fib(int n) {
return fib(n, 1, 1);
}
public int fib(int n, int first, int second){
if(n < 2) return n == 0 ? 0 :first;
return fib(n - 1, second % mod, (first + second) % mod);
}
}
未优化的动态规划 | 时间复杂度: O ( n ) O(n) O(n) | 空间复杂度 : O ( n ) O(n) O(n) | 推荐
class Solution {
int mod = 1000000007;
public int fib(int n){
int[] dp = new int[n + 1];
for(int i = 0; i < n + 1; i++){
if(i < 2){
dp[i] = i;
continue;
}
dp[i] = (dp[i - 1] + dp[i - 2]) % mod;
}
return dp[n];
}
}
优化的动态规划(空间压缩) | 时间复杂度: O ( n ) O(n) O(n)| 空间复杂度: O ( 1 ) O(1) O(1) | 推荐
class Solution {
int mod = 1000000007;
public int fib(int n){
int a = 0;
int b = 1;
int sum = 1;
for(int i = 0; i < n; i++){
sum = (a + b) % mod;
a = b;
b = sum;
}
return a;
}
}