hdu1575 Tr A 矩阵初识

A为一个方阵，则Tr A表示A的迹（就是主对角线上各项的和），现要求Tr(A^k)%9973。
Input数据的第一行是一个T，表示有T组数据。
每组数据的第一行有n(2 <= n <= 10)和k(2 <= k < 10^9)两个数据。接下来有n行，每行有n个数据，每个数据的范围是[0,9]，表示方阵A的内容。
Output对应每组数据，输出Tr(A^k)%9973。Sample Input

2
2 2
1 0
0 1
3 99999999
1 2 3
4 5 6
7 8 9

Sample Output

2
2686

一面赶期中考，一面刷水题，QwQ，感觉第一次大学考试药丸辣。。。

#include <bits/stdc++.h>
using namespace std;
#define LL long long
const int maxn=12;
int n;
struct mat
{
int  m[maxn][maxn];
mat(){memset(m,0,sizeof(m));};
mat friend operator*(mat a,mat b){
mat d;
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
for(int k=0;k<n;k++)
d.m[i][j]=(d.m[i][j]+a.m[i][k]*b.m[k][j]%9973)%9973;
return d;
}
};
mat pow(mat x,int cnt){
mat d;
for(int i=0;i<n;i++) d.m[i][i]=1;
while(cnt){
if(cnt&1) d=d*x;
x=x*x;
cnt>>=1;
}
return d;
}
int main()
{
int i,j,k,num,T;
scanf("%d",&T);
while(T--){
mat ans;
scanf("%d%d",&n,&k);
num=0;
for(i=0;i<n;i++)
for(j=0;j<n;j++)
scanf("%d",&ans.m[i][j]);
ans=pow(ans,k);
for(i=0;i<n;i++) num+=ans.m[i][i];
printf("%d\n",num%9973);
}
return 0;
}

View Code