斐波拉契数列的定义如下:
F(0) = 0, F(1) = 1 F(N) = F(N - 1) + F(N - 2), 其中 N > 1.
- 暴力递归
int fib(int n)
{
if (n == 0)
return 0;
if (n == 1 || n == 2)
return 1;
return fib(n-1) + fib(n-2);
}
我们画一个图来演示一下fib(5)的计算过程:

可以发现有很多重复计算,所以这种方式效率极低。
- 带记忆数组的递归
我们可以把f(1),f(2),f(3)这种子问题的返回值急着返回,先存到记忆数组里,在下次需要求解的时候直接先去记忆数组里查询就好了。
int fib(int n)
{
std::vector<int> memo(n + 1, 0);
if (n == 0)
{
memo[n] = 0;
}
if (n == 1 || n == 2)
{
memo[n] = 1;
}
if (n > 2)
{
if (memo[n] != 0)
return memo[n];
memo[n] = fib2(n - 1) + fib2(n - 2);
}
return memo[n];
}
通过递归我们“自顶向下”从一个规模较大的原问题,向下逐渐分解规模,最后直到分解成f(0),f(1),f(2)这样的base case,然后再逐层返回结果,这就是“自顶向下”。
但是我们也知道,f(3)是由f(1)和f(2)的结果推导出来的,也就是最简单的最小规模的问题往上推,最后是能够推出我们想要的较大规模的答案,这个就是动态规划的思想,通过循环迭代取代递归。
- 循环迭代
int fib3(int n)
{
std::vector<int> dp(n + 1, 0);
dp[0] = 0;
dp[1] = dp[2] = 1;
for (int i = 3; i <= n; i++)
dp[i] = dp[i - 1] + dp[i - 2];
return dp[n];
}
- 聊聊动态规划
动态规划的三个要素:1.存在重叠子问题。2.具备最优子结构。3.状态转移方程。
那么斐波拉契数列的状态转移方程就是:
f
(
n
)
=
1
,
n
=
1
,
2
;
f
(
n
?
1
)
+
f
(
n
?
2
)
,
n
>
2
;
f(n) = {1, n =1, 2;f(n-1) + f(n-2), n>2;}
f(n)=1,n=1,2;f(n?1)+f(n?2),n>2;
- 优化空间复杂度
从状态转移方程可以看出,f(n)只依赖之前的两个状态,因此我们只需要存储之前的两个状态,而不需要存储所有的中间状态。
int fib4(int n)
{
if (n == 0)
return 0;
if (n == 1 || n == 2)
return 1;
int pre = 1;
int prepre = 1;
for (int i = 3; i <= n; i++)
{
int sum = pre + prepre;
prepre = pre;
pre = sum;
}
return pre;
}
完整代码:
#include <iostream>
#include <vector>
int fib1(int n)
{
if (n == 0)
return 0;
if (n == 1 || n == 2)
return 1;
return fib1(n - 1) + fib1(n - 2);
}
int fib2(int n)
{
std::vector<int> memo(n + 1, 0);
if (n == 0)
{
memo[n] = 0;
}
if (n == 1 || n == 2)
{
memo[n] = 1;
}
if (n > 2)
{
if (memo[n] != 0)
return memo[n];
memo[n] = fib2(n - 1) + fib2(n - 2);
}
return memo[n];
}
int fib3(int n)
{
std::vector<int> dp(n + 1, 0);
dp[0] = 0;
dp[1] = dp[2] = 1;
for (int i = 3; i <= n; i++)
dp[i] = dp[i - 1] + dp[i - 2];
return dp[n];
}
int fib4(int n)
{
if (n == 0)
return 0;
if (n == 1 || n == 2)
return 1;
int pre = 1;
int prepre = 1;
for (int i = 3; i <= n; i++)
{
int sum = pre + prepre;
prepre = pre;
pre = sum;
}
return pre;
}
int main()
{
std::cout << fib1(30) << std::endl;
std::cout << fib2(30) << std::endl;
std::cout << fib3(30) << std::endl;
std::cout << fib4(30) << std::endl;
return 0;
}
|