矩阵乘法及其应用

定义

  • 对于两个矩阵 A(m*p),B(p*n),他们的乘积为 C,那么有:
  • Ci,j=k=1pAi,k+Bk,j
  • 看百度更清楚,点击进入百度百科界面

快速幂

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

应用

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

结论

  • 可以弄一个 2*2 的矩阵
  • A=[0111]
  • 那么会发现,斐波那契数列除前两个1外,第 n 项就是 An ,可以自己试试。

证明

  • 可以归纳证明

程序

//Luogu-1962
#include 
#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]) ;
}

你可能感兴趣的:(知识)