矩阵快速幂和普通快速幂写法相似。
struct Matrix
{
int maze[maxn][maxn];
void Init()
{
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
maze[i][j]=(i==j);
}
friend Matrix operator*(Matrix a,Matrix b)
{
Matrix c;
memset(c.maze,0,sizeof c.maze);
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
for(int k=0;k<n;k++)
c.maze[i][j]=(c.maze[i][j]+a.maze[i][k]*b.maze[k][j])%MOD;
return c;
}
friend Matrix operator^(Matrix a,int k)
{
Matrix c;
c.Init();
while(k)
{
if(k&1)
c=c*a;
a=a*a;
k>>=1;
}
return c;
}
};
水题,练习下hdu 1575