矩阵快速幂求Fibonacci数列

小白选手又来更新算法了

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

从我们熟悉的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初始化的系数矩阵即

memset初始化的结果矩阵即
对于Fibonacci数列的快速幂求法如下：

#include <iostream>
#include <cstring>
#include <string>
#include <cstdio>

using namespace std;

const int mod = 1e9 + 7;

//matrix n.矩阵
struct mat
{
    //五行五列二维数组(数组开的大一些)
    long long a[5][5];
};

//两个矩阵相乘,返回矩阵
mat mul(mat a,mat b)
{
    mat ans;//结果
    memset(ans.a,0,sizeof ans.a);//初始化:ans为mat类型的结构体,ans.a为矩阵
    
    for (int i = 0; i < 2;i++)
        for (int j = 0;j < 2;j++)
            for (int k = 0;k < 2;k++)
                ans.a[i][j] += (a.a[i][k] * b.a[k][j]) % mod;//a和b为mat类型的结构体,a.a和b.a为矩阵
    
    return ans;
}

//n代表指数
long long quick_power(long long n)
{
    mat ans,res;//ans为结果矩阵,res为系数矩阵
    
    memset(ans.a,0,sizeof ans.a);//初始化结果矩阵
    ans.a[0][0] = 1;
    ans.a[1][0] = 1;
    
    memset(res.a,0,sizeof res.a);//初始化系数矩阵
    res.a[0][0] = 1;
    res.a[0][1] = 1;
    res.a[1][0] = 1;
    res.a[1][1] = 0;
    
    //下面是普通快速幂
    while (n)
    {
        if (n & 1) ans = mul(res,ans);//res和ans两个矩阵不能颠倒
        n >>= 1;//指数减半
        res = mul(res,res);//底数平方
    }
    return ans.a[0][0];//ans.a[0][0]即图中的f(n)
}

int main()
{
    int n;
    while (cin >> n)
    {
        //1 1 2 3 5 8 13 21 34
        if (n == 1 || n == 2) cout << 1 << endl;
        else cout << quick_power(n - 2) << endl; 
    }
    return 0;
}