矩阵快速幂模板

矩阵快速幂跟快速幂求模原理一样,只不过在把乘号换成矩阵相乘。

然后res换成单位矩阵( 即主对角线的元素为一)

它能快速求递推式,把O(n)降为O(logn)

模板如下:

 1 #define mod 1000000007
 2 typedef long long ll;
 3 #define MAX 105
 4 const int N=3;//矩阵的大小
 5 int T;
 6 ll n;
 7 struct hh
 8 {
 9     ll ma[N][N];
10 }a,res;
11 hh multi(hh a,hh b) 12 { 13 hh tmp; 14 memset(tmp.ma,0,sizeof(tmp.ma)); 15 for(int i=0;i<N;i++) 16 for(int j=0;j<N;j++) 17 for(int k=0;k<N;k++) 18 { 19 tmp.ma[i][j]=add(tmp.ma[i][j],a.ma[i][k]*b.ma[k][j]); 20 } 21 return tmp; 22 } 23 void fast_pow(hh a,long long k) 24 { 25 memset(res.ma,0,sizeof(res.ma)); 26 for(int i=0;i<N;i++)res.ma[i][i]=1; 27 while(k>0) 28 { 29 if(k&1) res=multi(res,a); 30 a=multi(a,a); 31 k>>=1; 32 } 33 }

 下面的链接也可以看看:

https://www.jianshu.com/p/25eba927d9da

https://blog.csdn.net/aiwen1413/article/details/53445064

posted @ 2018-07-27 17:34  better46  阅读(387)  评论(0编辑  收藏  举报