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

矩阵快速幂 模板

时间:2016-08-06 16:03:24      阅读:155      评论:0      收藏:0      [点我收藏+]

标签:

#define Matr 10 //矩阵大小,注意能小就小  
struct mat//矩阵结构体,a表示内容,size大小 矩阵从1开始  
{  
    ll a[Matr][Matr],size;  
    mat()  
    {  
        size=0;  
        memset(a,0,sizeof(a));  
    }  
};  
void print(mat m)//输出矩阵信息,debug用   
{  
    int i,j;  
    printf("%d\n",m.size);  
    for(i=0;i<m.size;i++)  
    {  
        for(j=0;j<m.size;j++)printf("%d ",m.a[i][j]);  
        printf("\n");  
    }  
}  
  
mat multi(mat m1,mat m2,int mod)//两个相等矩阵的乘法,对于稀疏矩阵,有0处不用运算的优化   
{  
    mat ans=mat();    ans.size=m1.size;  
    for(int i=1;i<=m1.size;i++)  
        for(int j=1;j<=m2.size;j++)  
            if(m1.a[i][j])//稀疏矩阵优化   
                for(int k=1;k<=m1.size;k++)  
                    ans.a[i][k]=(ans.a[i][k]+m1.a[i][j]*m2.a[j][k])%mod;  
  
    return ans;  
}  
mat quickmulti(mat m,int n,int mod)//二分快速幂   
{  
    mat ans=mat();  
    int i;  
    for(i=1;i<=m.size;i++)ans.a[i][i]=1;  
    ans.size=m.size;  
    while(n)  
    {  
        if(n&1)ans=multi(m,ans,mod);  
        m=multi(m,m,mod);  
        n>>=1;  
    }  
    return ans;  
}  

 

矩阵快速幂 模板

标签:

原文地址:http://www.cnblogs.com/shimu/p/5744134.html

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