矩阵快速幂的应用,先定义一个结构体,定义矩阵和矩阵乘法,方便以后直接调用,再写一个矩阵快速幂的函数,进行相关运算。其实对于比较大的指数运算时,快速幂的优势是很明显的,时间复杂度降低到很小。
#include <iostream>
#include <fstream>
#include <memory.h>
#define mode 9973
using namespace std;
int n;
struct matrix
{
int mat[11][11];
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] % mode;
tmp.mat[i][j] %= mode;
}
}
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] % mode;
sum %= mode;}
return sum;
}
int main()
{
//ifstream cin("in.txt");
matrix cat;
int T,k;cin >> T;
while(T --)
{
cin >> n >> k;
for(int i = 0;i < n;i ++)
for(int j = 0;j < n;j ++)
cin >> cat.mat[i][j];
cout << pow(cat,k) << endl;
}
return 0;
}