【数论】矩阵快速幂，递推优化，模板

目录

一，矩阵快速幂用于优化递推

二，矩阵快速幂的推导

---

一，矩阵快速幂用于优化递推

矩阵快速幂用于优化递推公式，例如：斐波那契的递推公式为 f[1] = 1,f[2] = 1, f[n] = f[n - 1] + f[n - 2](n >= 3)。当我们想要求第1e8项时，直接暴力求的时间复杂度为o（n），需要进行1e8次运算，显然会超时，于是需要用到矩阵快速幂来优化。

二，矩阵快速幂的推导

斐波那契的递推公式为f[n] = f[n - 1] + f[n - 2]，我们可以用线性代数中矩阵相乘的形式来表示这个公式。如下

进一步可以推出

这时候我们只需求出矩阵的n - 2次方，再求出第一行与矩阵的乘积和便可以求出f(n)的值，求矩阵的幂时用快速幂的方法，时间复杂度便可以优化到log级别。光用文字叙述可能有点难懂，结合代码看一下清楚很多。

具体代码如下

#include<bits/stdc++.h>
using namespace std;
const int N = 10;
const int n = 2;
int x;

int mul(int a[][N], int b[][N], int p) //矩阵a * b，并把值存到a中;
{
    int temp[N][N];
    memset(temp, 0, sizeof temp);
    for(int i = 1; i <= n; i++)
        for(int j = 1; j <= n; j++)
            for(int k = 1; k <= n; k++)
            {
                temp[i][j] = (temp[i][j] + a[i][k] * b[k][j]) % p;
            }
    for(int i = 1; i <= n; i++)
        for(int j = 1; j <= n; j++)
        {
            a[i][j] = temp[i][j];
        }
}

int qmi(int a[][N], int k) //快速幂求矩阵a的k次方
{
    int res[N][N];
    memset(res, 0, sizeof res);
    for(int i = 1; i <= n; i++)
    {
        res[i][i] = 1;
    }
    
    while(k)
    {
        if(k & 1)mul(res, a, m);
        k >>= 1;
        mul(a, a, m);
    }
    
    for(int i = 1; i <= n; i++)
        for(int j = 1; j <= n; j++)
        {
            a[i][j] = res[i][j];
        }
}
void solve()
{
    int b[N][N] = {  //因为我是从1开始遍历，所以多加了一层0;
        {0, 0, 0},
        {0, 1, 1},
        {0, 1, 0}
    };
    int a[N][N] = {
        {0, 0},
        {0, 1},
        {0, 1}
    };
    qmi(b, x - 2);
    int ans = 0;
    for(int i = 1; i <= 2; i++)
    {
        ans = (ans + b[1][i] * a[i][1]) % m;
    }
    cout << ans << endl;
}
signed main()
{
    while(cin >> x)
    {
        solve();
    }
}