结构体类型快速驱魔运算及运算符的重载

struct Matrix
{
long long mat

;
Matrix operator*(const Matrix m)const//定义矩阵乘法的运算符*
{
Matrix tmp;
for(int i = 0;i < n;i++)
{
for(int j = 0;j < n;j++)
{
tmp.mat[i][j] = 0;
for(int k = 0;k < n;k++)
{
tmp.mat[i][j] += mat[i][k]*m.mat[k][j]%MOD;//这里需要重点掌握
tmp.mat[i][j] %= MOD;
}
}
}
return tmp;
}
};
long long Pow(Matrix &m,int k)
{
Matrix ans;
memset(ans.mat,0,sizeof(ans.mat));
for(int i = 0;i < n;i++)
ans.mat[i][i] = 1;
while(k)
{
if(k&1) ans = ans*m;//这里是以二进制形式转化的，需要细细体会
k >>= 1;
m = m*m;
}
long long sum = 0;
for(int i = 0;i < n;i++)
{
sum += ans.mat[i][i]%MOD;//这里需要注意，这里和先把所有数据加起来再取模是等价的。
sum %= MOD;
}
return sum;
}