看了好几篇矩阵快速幂的相关入门资料,写的公式漫天飞的,眼睛都花了也没看懂,为了防止这种事再次发生,特把相关心得记下,下次方便查阅
矩阵快速幂=快速幂+矩阵乘法
矩阵快速幂就是把快速幂的函数内容换成3层循环的矩阵乘法
#pragma GCC optimize("Ofast","inline","-ffast-math")
#pragma GCC target("avx,sse2,sse3,sse4,mmx")
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N = 2e5+10;
const int mod=1000000007;
int n,k;
struct node
{
int a[110][110];
node()
{
memset(a,0,sizeof(a));
}
inline void build()
{
for(int i=0; i<n; ++i)
a[i][i]=1;
}
} a,ans;
node operator *(const node &x,const node &y)
{
node z;
for(int i=0; i<n; ++i)
for(int j=0; j<n; ++j)
for(int k=0; k<n; ++k)
z.a[i][j]=(z.a[i][j]+x.a[i][k]*y.a[k][j]%mod)%mod;
return z;
}
signed main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cin>>n>>k;
for(int i=0; i<n; ++i)
for(int j=0; j<n; ++j)
cin>>a.a[i][j];
ans.build();
while(k)
{
if(k&1)
ans=ans*a;
k>>=1;
a=a*a;
}
for(int i=0; i<n; ++i)
{
for(int j=0; j<n; ++j)
cout<<ans.a[i][j]<<" ";
cout<<endl;
}
return 0;
}
P.S.重载运算符之后内容就基本上可以写的和快速幂一样了