首页 > 代码库 > 矩阵快速幂模板

矩阵快速幂模板

typedef vector<ll>vec;
typedef vector<vec>mat;
mat mul(mat &A,mat &B)
{
mat C(A.size(),vec(B[0].size()));
for(ll i=0;i<A.size();i++)
{
for(ll k=0;k<B.size();k++)
{
for(ll j=0;j<B[0].size();j++)
C[i][j]=(C[i][j]+A[i][k]*B[k][j]%mod)%mod;
}
}
return C;
}
mat pow(mat A,ll n)
{
mat B(A.size(),vec(A.size()));
for(int i=0;i<A.size();i++)
B[i][i]=1;
while(n>0)
{
if(n%2)
B=mul(B,A);
A=mul(A,A);
n/=2;
}
return B;
}

矩阵快速幂模板