Problem Description
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的内容。
每组数据的第一行有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
22686
#include<cstdio> #include<queue> #include<vector> #include<map> #include<iostream> #include<algorithm> #include<functional> using namespace std; const int maxn=100005; const int size=15; const int base=9973; int a[size][size]; int b[size][size]; int c[size][size]; int d[size][size]; int n,k,T; void get(int x) { memset(c,0,sizeof(0)); memcpy(b,a,sizeof(c)); for (int i=x,j=1;i;i>>=1) { if (i&1) if (j) {j=0; memcpy(c,b,sizeof(c));} else { memcpy(d,c,sizeof(c)); memset(c,0,sizeof(c)); for (int i1=1;i1<=n;i1++) for (int i2=1;i2<=n;i2++) for (int i3=1;i3<=n;i3++) (c[i1][i3]+=b[i1][i2]*d[i2][i3])%=base; } memset(d,0,sizeof(d)); for (int i1=1;i1<=n;i1++) for (int i2=1;i2<=n;i2++) for (int i3=1;i3<=n;i3++) (d[i1][i3]+=b[i1][i2]*b[i2][i3])%=base; memcpy(b,d,sizeof(d)); } memcpy(a,c,sizeof(c)); } int main() { cin>>T; while (T--) { cin>>n>>k; int ans=0; for (int i=1;i<=n;i++) for (int j=1;j<=n;j++) scanf("%d",&a[i][j]); get(k); for (int i=1;i<=n;i++) (ans+=a[i][i])%=base; cout<<ans<<endl; } return 0; }