矩阵乘法及其应用

定义

  • 对于两个矩阵 A(m*p),B(p*n),他们的乘积为 C,那么有:
  • Ci,j=k=1pAi,k+Bk,j
    <script type="math/tex; mode=display" id="MathJax-Element-1"> C_{i,j}=\sum_{k=1}^{p} A_{i,k}+B_{k,j} </script>
  • 看百度更清楚,点击进入百度百科界面

快速幂

  • 可以做个函数返回两个矩阵的积,那么利用快速幂的思想,就可以在log的时间内求出一个矩阵的幂。
  • 实际上除了乘的部分需要稍微改一下,其它都相同,注意一下初始化就行了。
  • 初始化ksm()ret的时候,需要初始化成单位矩阵,即 a[i][i]=1,其它为0的矩阵。
  • 之前写过一篇快速幂的博文传送门
  • 模板题 http://blog.csdn.net/jackypigpig/article/details/78453575

应用

快速求斐波那契数列某一项

结论

  • 可以弄一个 2*2 的矩阵
  • A=[0111]
    <script type="math/tex; mode=display" id="MathJax-Element-2"> A= \begin{bmatrix} 0 & 1\\ 1 & 1 \end{bmatrix} </script>
  • 那么会发现,斐波那契数列除前两个1外,第 n 项就是 An <script type="math/tex" id="MathJax-Element-3">A^n</script>,可以自己试试。

证明

  • 可以归纳证明

程序

//Luogu-1962
#include <cstdio>
#define Ha 1000000007
typedef long long ll;
struct matrix{ll a[2][2];} A;
ll n;

matrix multi(matrix x,matrix y){
    matrix z={
   
   0};
    for (ll i=0; i<=1; i++)
        for (ll j=0; j<=1; j++)
            for (ll k=0; k<=1; k++)
                z.a[i][j]=(z.a[i][j]+(x.a[i][k]*y.a[k][j])%Ha)%Ha;
    return z;
}

matrix ksm(matrix x,ll y){
    matrix ret={
   
   0};
    ret.a[0][0]=ret.a[1][1]=1;
    for (; y; y>>=1,x=multi(x,x))
        if (y&1) ret=multi(ret,x);
    return ret;
}

int main(){
    A.a[0][0]=0,A.a[0][1]=A.a[1][0]=A.a[1][1]=1;    
    scanf("%lld",&n);
    printf("%lld\n",ksm(A,n-1).a[1][1]) ;
}

猜你喜欢

转载自blog.csdn.net/jackypigpig/article/details/78363631