矩阵快速幂的大体思想与快速幂想法一致,只是普通的数乘,变成了矩阵相乘,所以问题的关键在于矩阵乘法上,即(看代码)
ll pow(ll x,ll n,ll mod){ ll res=1;//此处的1变为单位阵 while(n>0){ if(n&1)res=res*x%mod;//res*x变成矩阵乘法 x=x*x%mod;//同上 n>>=1; } return res; }
矩阵乘法板子(个人认为这是矩阵快速幂的关键):
typedef vector<int> vec; typedef vector<vec> mat; mat mul(mat &A,mat&B){ mat C(A.size(),vec(B[0].size())); for(int i=0;i<A.size();i++){ for(int k=0;k<B.size();k++){ for(int j=0;j<B[0].size();j++){ C[i][j]=(C[i][j]+A[i][k]*B[k][j])%MOD; } } } return C; }
矩阵快速幂(核心):
mat pow(mat A,ll n){ mat B(A.size(),vec(A.size())); //单位阵 I for(int i=0;i<A.size();i++){ B[i][i]=1; } while(n>0){ if(n&1)B=mul(B,A); A=mul(A,A); n>>=1; } return B; }