矩阵快速幂求Fibonacci数列

最新推荐文章于 2022-09-08 14:37:35 发布

原创

最新推荐文章于 2022-09-08 14:37:35 发布 · 3k 阅读

本文介绍了一种利用矩阵快速幂算法高效计算斐波那契数列的方法。通过将斐波那契数列的递推公式转化为矩阵的形式，并使用快速幂算法加速计算过程。文中详细解释了矩阵快速幂的原理，并给出了具体的C++实现代码。

小白选手又来更新算法了

矩阵快速幂算法是以快速幂算法为基础的

从我们熟悉的Fibonacci数列入手，设一个Fibonacci数列为1，1，2，3，5，8，13，21… …可以得到其递推公式为f(n) = f(n - 1) + f(n - 2)

如果用矩阵(线性代数中矩阵乘法)来表示其递推公式：
在这里插入图片描述
同理，将等式左边换成

中间的系数矩阵不变，则等式右边相应的变为在这里插入图片描述
易得递推式并发现规律（粉色+蓝色=红色）

此时，将问题转化为了系数矩阵的幂次问题。

我们用二维数组来存储矩阵，设两个矩阵 int A[m][n]，int B[n][p]，再建立一个答案矩阵 int C[m][p]。
可通过如下代码实现两个矩阵的乘法运算

for(int i = 0;i < m;i++)
    for (int k = 0;k < n;k++)
        for (int j = 0;j < p;j++)
            C[i][j] += A[i][k] * B[k][j];

A[m][n] * B[n][p] = C[m][p]

我们把矩阵用一个struct封装起来并重载运算符，再套上一个普通快速幂的模板，就可以实现矩阵的快速幂及取模
注意：
下面代码中memset初始化的系数矩阵即