题目链接

牛客网

题目描述

求斐波那契数列的第 n 项,n <= 39。
image.png

解题思路

如果使用递归求解,会重复计算一些子问题。例如,计算 f(4) 需要计算 f(3) 和 f(2),计算 f(3) 需要计算 f(2) 和 f(1),可以看到 f(2) 被重复计算了。
image.png

递归是将一个问题划分成多个子问题求解,动态规划也是如此,但是动态规划会把子问题的解缓存起来,从而避免重复求解子问题。

  1. class Solution {
  2. public:
  3. const int mod = 1000000007;
  4. int Fibonacci(int n) {
  5. if(n<2)
  6. return n;
  7. return Fibonacci(n-1)%mod+Fibonacci(n-2)%mod;
  8. }
  9. };

考虑到第 i 项只与第 i-1 和第 i-2 项有关,因此只需要存储前两项的值就能求解第 i 项,从而将空间复杂度由 O(N) 降低为 O(1)。
由于待求解的 n 小于 40,因此可以将前 40 项的结果先进行计算,之后就能以 O(1) 时间复杂度得到第 n 项的值。

  1. class Solution {
  2. private:
  3. int fib[40];
  4. public:
  5. Solution(){
  6. fib[1] = 1;
  7. for (int i = 2; i < 40; i++)
  8. fib[i] = fib[i - 1] + fib[i - 2];
  9. }
  10. int Fibonacci(int n) {
  11. return fib[n];
  12. }
  13. };
  • 时间复杂度:O(1)
  • 空间复杂度:O(1)