码迷,mamicode.com
首页 > 其他好文 > 详细

矩阵快速幂

时间:2018-01-29 20:15:48      阅读:153      评论:0      收藏:0      [点我收藏+]

标签:一个   矩阵乘法   ima   mod   pow   ica   img   mat   bubuko   

 

矩阵快速幂讲解

若求斐波那契数列第n项的数值,我们可以递推求得,若n值(>1e10)很大,我们一个一个求,一定会超时,下面讲解一下用矩阵快速幂求解

矩阵快速幂就是求矩阵A^n的矩阵,如同快速幂一样,代码极其相似,将快速幂中的整数乘法改为矩阵的乘法即可

矩阵乘法:                                                                                                                                                                                                                                                                       技术分享图片                                                 

其中c[i][j]为矩阵A的第i行与矩阵B的第j列对应想乘的和,即

矩阵乘法代码实现如下:

技术分享图片
Matrix multiplicative(Matrix a,Matrix b){
    Matrix res;
    memset(res.matrix,0,sizeof(res.matrix));
    for(int i = 0 ; i < n ; i++)
        for(int j = 0 ; j < n ; j++)
            for(int k = 0 ; k < n ; k++)
                res.matrix[i][j] += a.matrix[i][k]*b.matrix[k][j];
    return res;
}
View Code

矩阵快速幂代码如下:

技术分享图片
Matrix pow(Matrix mx,int m){
    Matrix res,base=mx;
    init(res); //初始为单位矩阵,即除主对角线都是1外,其他都是0
    while(m)
    {
        if(m&1)
            res=multiplicative(res,base);
        base=multiplicative(base,base);
        m>>=1;
    }
    return res;
}
View Code

斐波那契数列递推式为F(n)=F(n-1)+F(n-2) (n>=2)

F(0)=0,F(1)=1 

根据递推式我们可以列出两个方程:F(n)=F(n-1)+F(n-2)和F(n-1)=F(n-2)+F(n-3)

由此

技术分享图片

技术分享图片

代码如下:

技术分享图片
#include<stdio.h>
#include<cstring>
#define LL long long
#define M 10
#define MOD 1000000007
struct Matrix{
    LL matrix[M][M];
};
int n;//矩阵的阶数 
void init(Matrix &res){
    for(int i=0;i<=n;i++)
    {
        for(int j=0;j<=n;j++)
            res.matrix[i][j]=0;
        res.matrix[i][i]=1;
    }
}
Matrix multiplicative(Matrix a,Matrix b){
    Matrix res;
    memset(res.matrix,0,sizeof(res.matrix));
    for(int i = 0 ; i < n ; i++)
        for(int j = 0 ; j < n ; j++)
            for(int k = 0 ; k < n ; k++)
                res.matrix[i][j] =(res.matrix[i][j]%MOD+a.matrix[i][k]%MOD*b.matrix[k][j]%MOD)%MOD;
    return res;
}
Matrix pow(Matrix mx,int m){
    Matrix res,base=mx;
    init(res); //初始为单位矩阵,即除主对角线都是1外,其他都是0
    while(m)
    {
        if(m&1)
            res=multiplicative(res,base);
        base=multiplicative(base,base);
        m>>=1;
    }
    return res;
}
int main(){
    int m;
    n=2;
    scanf("%d",&m);
    Matrix base,res;
    base.matrix[0][0]=base.matrix[0][1]=base.matrix[1][0]=1;
    base.matrix[1][1]=0;
    res=pow(base,m-1);
    if(m==0)
        puts("0");
    else if(m==1)
        puts("1");
    else
        printf("%lld\n",res.matrix[0][0]%MOD);
    return 0;    
} 
View Code

 

常见递推类型:

技术分享图片

 

矩阵快速幂

标签:一个   矩阵乘法   ima   mod   pow   ica   img   mat   bubuko   

原文地址:https://www.cnblogs.com/lemon-jade/p/8378880.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!