今天来给大家介绍几个在面试中常用的、求解斐波那契数列问题的方法。
首先定义斐波那契数列问题:
定义 $a_0 = 1$, $a_1 = 1$, $a_n = a_{n-1} + a_{n-2}$,求 $a_n$ 是多少。
为了避免考虑整数溢出问题,我们求 $a_n \% p$ 的值,$p=10^9+7$。
算法1
递归。
递归计算的节点个数是 $O(2^n)$ 的级别的,存在大量重复计算。
时间复杂度是 $O(2^n)$,一秒内大约能算到第三四十项。
C++ 代码
const int MOD = 1000000007;
int f(int n)
{
if (n <= 1) return 1;
return (f(n - 1) + f(n - 2)) % MOD;
}
算法2
记忆化搜索。
开一个大数组记录中间结果,如果一个状态被计算过,则直接查表,否则再递归计算。
总共有 $n$ 个状态,计算每个状态的复杂度是 $O(1)$,所以时间复杂度是 $O(n)$。
一秒内算 $n=10^7$ 毫无压力,但由于是递归计算,递归层数太多会爆栈,大约只能算到 $n=10^5$ 级别。
C++ 代码
const int N = 100000, MOD = 1000000007;
int a[N];
int f2(int n)
{
if (a[n]) return a[n];
if (n <= 1) return 1;
a[n] = f2(n - 1) + f2(n - 2);
a[n] %= MOD;
return a[n];
}
算法3
递推。
开一个大数组,记录每个数的值。用循环递推计算。
总共计算 $n$ 个状态,所以时间复杂度是 $O(n)$。
但需要开一个长度是 $n$ 的数组,内存将成为瓶颈,当 $n=10^8$ 时,需要的内存是 $\frac{4 * 10^8}{1024 \times 1024} \approx 381MB$。
分子中乘4是因为C++中 int
类型占4字节。
C++代码
const int N = 100000000, MOD = 1000000007;
int f3(int n)
{
a[0] = a[1] = 1;
for (int i = 2; i <= n; i ++ )
{
a[i] = a[i - 1] + a[i - 2];
a[i] %= MOD;
}
return a[n];
}
算法4
递归+滚动变量。
仔细观察我们会发现,递推时我们只需要记录前两项的值即可,没有必要记录所有值,所以我们可以用滚动变量递推。
时间复杂度还是 $O(n)$,但空间复杂度变成了 $O(1)$。
C++代码:
const int MOD = 1000000007;
int f4(int n)
{
if (n <= 1) return 1;
int x, y, z;
x = y = 1;
for (int i = 2; i <= n; i ++ )
{
z = (x + y) % MOD;
x = y;
y = z;
}
return z;
}
算法5
矩阵运算 + 快速幂。
快速幂算法的模板可以参考这里。
用算法4我们1秒内最多可以算到 $10^8$ 级别,那当 $n$ 更大时该怎么办呢?
可以先利用矩阵运算的性质将通项公式变成幂次形式,然后用平方倍增(快速幂)的方法求解第 $n$ 项。
首先我们定义向量
$$
X_n = \begin{array}{cc}
[a_n & a_{n-1}]
\end{array}
$$
边界:
$$
X_1 = \begin{array}{cc}
[a_1 & a_0]
\end{array}
$$
然后我们可以找出矩阵:
$$
A = \left [ \begin{array}{cc}
1 & 1 \\
1 & 0
\end{array} \right ]
$$
则有:
$$
X_n = X_{n-1} \times A
$$
所以:
$$ X_n = X_1 \times A^{n-1} $$
由于矩阵具有结合律,所以我们可以先求出 $A^{n-1} \% P$,然后再用 $X_1$ 左乘,即可求出 $X_n$,向量 $X_n$ 的第一个元素就是 $a_n$。
时间复杂度分析:快速幂的时间复杂度是 $O(logn)$,所以算法5的时间复杂度也是 $O(logn)$。
C++代码
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <ctime>
using namespace std;
const int MOD = 1000000007;
void mul(int a[][2], int b[][2], int c[][2])
{
int temp[][2] = {{0, 0}, {0, 0}};
for (int i = 0; i < 2; i ++ )
for (int j = 0; j < 2; j ++ )
for (int k = 0; k < 2; k ++ )
{
long long x = temp[i][j] + (long long)a[i][k] * b[k][j];
temp[i][j] = x % MOD;
}
for (int i = 0; i < 2; i ++ )
for (int j = 0; j < 2; j ++ )
c[i][j] = temp[i][j];
}
int f_final(long long n)
{
if (!n) return 1;
int x[2] = {1, 1};
int res[][2] = {{1, 0}, {0, 1}};
int t[][2] = {{1, 1}, {1, 0}};
long long k = n - 1;
while (k)
{
if (k&1) mul(res, t, res);
mul(t, t, t);
k >>= 1;
}
int c[2] = {0, 0};
for (int i = 0; i < 2; i ++ )
for (int j = 0; j < 2; j ++ )
{
long long r = c[i] + (long long)x[j] * res[j][i];
c[i] = r % MOD;
}
return c[0];
}
int main()
{
long long n ;
cin >> n;
cout << f_final(n) << endl;
return 0;
}
算法4有一丁点的小问题,就是当n == 0 || n == 1的时候 z 木有值
然后算法5 牛逼!
多谢指正,代码已修正。
算法4是递推+滚动变量吧
对哦
记忆化搜索是不是有点动态规划的思想在里头
记忆化搜索是dp的一个分支
为什么暴力递归的时间复杂度是O(n)啊?
算法3的代码中缺少了 int a[N];
int x[2] = {1, 1};
int res[][2] = {{1, 0}, {0, 1}};
这个初值是什么意思?
orzorzorzorz,收藏了!
算法5 那个 int x[2] = {1, 1};是不是有点问题呀 不应该是 int x[2] = {1, 0};嘛 按照上面的算 f_final(2)就等于2了
感觉老师这里的初始值是这样的f[1] = f[0] = 1
记忆化搜索,数组好像也开不了太大
最后一步 $ x_n=x_1 ∗ A^{n−1} $,代码里 x[2] 为啥不等于 {1, 0}
看定义是 a[1] = a[0] = 1
再分享一种方法,更简单高效的方法,就是直接把菲波那切数列的通项公式解出来。
int Fibonacci(int n) {
double sqrt5 = sqrt(5);
double alpha = (1 + sqrt5) / 2;
double beta = (1 - sqrt5) / 2;
double ret = (pow(alpha, n) - pow(beta, n)) / sqrt5;
return (int) ret;
}
但是精度不高啊
还好,是可以ac的,只是n不能太大
万一公式记错了怎么办
学一下特征方程 再现场推
简单是简单,但是没有高效吧
1. 从时间复杂度来说:pow即使是快速幂也是log(n)复杂度,没有变
2. double的精度没有很高
3. 如何%p?代码中好像没有看到任何关于取模p的地方
tql
hh,怎么感觉算法4应该是递推+滚动变量
AcWing 1213 怎么做?
算法5
f_final
里面的a[2][2]
好像是多余的是的hh,已经去掉了
@y总,算法4应该是 x = 0, y =1;
x和y分别表示第0项和第1项,这里统一定义为1。
时间复杂度O(logn)
是指 以 10 为底 n 的对数吗?
如果深究应该是以2为底,但是时间复杂度这个概念不在乎系数,比如n和2n都是O(n)。由于对数运算的性质$log_{a}^{b}✖log_{b}^{n}=log_{a}^{n}$所以一般不再说以什么为底,就是O(logn)
$log_{a}^{b}$是个常数
哦,好的,明白
算法时间复杂度为log(n)时,不同底数对应的时间复杂度的倍数关系为常数,不会随着底数的不同而不同,因此可以将不同底数的对数函数所代表的时间复杂度,当作是同一类复杂度处理
对滴。算法复杂度中一般以2为底。
y中 这一步Xn=X1×An−1 对应的代码 long long r = c[i] + (long long)x[j] * res[i][j]; 这个res[i][j] 的i和j是不是反了 变成行乘行了 应该行乘列才对吧
多谢指正~ 已更正。之前$X$恰好是$[1, 1]$ 且 $res$ 恰好是对称矩阵,所以答案恰好是对的,但过程是错的。
y总,返回的c[0]好像是第 n + 1项,c[1]才是第 n 项
1 1 2 3 5 8
n = 5 时得到的 是 8
最前面的定义是:
所以 $a_5$ 的值在这种定义下就是8。