矩阵快速幂优化线性递推

我们熟知的斐波那契数列递推公式是:

\(f(n)=f(n-1)+f(n-2)\)

假设我们需要求斐波那契数列的第n项,当n非常大(如n=1e9)的时候,递推肯定超时。我们不妨设:

\(\binom{f_{n}}{f_{n-1}}=\begin{pmatrix}a & b\\ c & d\end{pmatrix}\binom{f_{n-1}}{f_{n-2}}\)

将等式右边乘开,得到: 

\(\binom{af_{n-1}+bf_{n-2}}{cf_{n-1}+df_{n-2}}\)

要使其等于等式左边,显然:

\(\binom{f_{n}}{f_{n-1}}=\begin{pmatrix}1 & 1\\ 1 & 0\end{pmatrix}\binom{f_{n-1}}{f_{n-2}}\)

如此递推下去,我们最终能够得到:

\(\binom{f_{n}}{f_{n-1}}=\begin{pmatrix}1 & 1\\ 1 & 0\end{pmatrix}^{n-2}\binom{f_{2}}{f_{1}}\)

我们已知\(f1=1,f2=1\),这样,我们就可以通过矩阵快速幂来求数列第n项了! 像这样,对于所有的线性递推关系,我们都可以使用矩阵快速幂来优化ovo

模版如下(洛谷P3390 矩阵快速幂):

#include <iostream>
#include <cstdio>
#include <cstring>
#define mod 1000000007
#define ll long long
using namespace std;

struct Matrix
{
    ll m[101][101];
}a,e;
ll n,k;

Matrix mul(Matrix x,Matrix y)
{
    Matrix c;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            c.m[i][j]=0;
        }
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            for(int k=1;k<=n;k++)
            {
                c.m[i][j]=c.m[i][j]%mod+x.m[i][k]*y.m[k][j]%mod;
            }
        }
    }
    return c;
}

Matrix mqpow(Matrix x,ll p)
{
    Matrix ans=e;
    while(p!=0)
    {
        if(p%2)
        {
            ans=mul(ans,x);
        }
        x=mul(x,x);
        p/=2;
    }
    return ans;
}

int main()
{
    scanf("%lld%lld",&n,&k);
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            scanf("%lld",&a.m[i][j]);
        }
    }
    for(int i=1;i<=n;i++)
    {
        e.m[i][i]=1;
    }//初始化
    Matrix ans=mqpow(a,k);
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            printf("%lld ",ans.m[i][j]%mod);
        }
        printf("\n");
    }

    return 0;
}

 

posted @ 2019-05-08 00:12  Black_Gzombie  阅读(361)  评论(0编辑  收藏  举报