题目描述
A为一个方阵,则Tr A表示A的迹(就是主对角线上各项的和),现要求Tr(A^k)%9973。
输入
数据的第一行是一个T,表示有T组数据。
每组数据的第一行有n(2 <= n <= 10)和k(2 <= k < 10^9)两个数据。接下来有n行,每行有n个数据,每个数据的范围是[0,9],表示方阵A的内容。
输出
对应每组数据,输出Tr(A^k)%9973。
样例输入
2
2 2
1 0
0 1
3 99999999
1 2 3
4 5 6
7 8 9
样例输出
2
2686
思路
直接矩阵快速幂
代码
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#include <math.h>
#define mem(a) memset(a,0,sizeof(a))
using namespace std;
int mod;
int n;
struct matrix
{
int a[12][12];
int r;
int c;
};
matrix multiply (matrix x,matrix y)
{
matrix t;
t.r=x.r;
t.c=y.c;
for(int i=0; i<x.r; ++i)
for(int j=0; j<y.c; ++j)
t.a[i][j]=0;
for(int i=0; i<x.r; ++i)
for(int j=0; j<y.c; ++j)
for(int k=0; k<x.r; ++k)
{
t.a[i][j]+=x.a[i][k]*y.a[k][j];
t.a[i][j]%=mod;
}
return t;
}
matrix powmatrix (matrix x,int y)
{
matrix t=x;
for(int i=0; i<n; ++i)
for(int j=0; j<n; ++j)
if(i==j)
t.a[i][j]=1;
else
t.a[i][j]=0;
while(y)
{
if(y&1)
t=multiply(t,x);
y/=2;
x=multiply(x,x);
}
return t;
}
int main ()
{
#ifndef ONLINE_JUDGE
freopen("1","r",stdin);
#endif // ONLINE_JUDGE
ios_base::sync_with_stdio(false);
cin.tie(0);
int k,T;
matrix t,x;
int ans;
cin>>T;
mod = 9973;
while(T--)
{
cin>>n>>k;
x.c=n;
x.r=n;
for(int i=0; i<n; ++i)
for(int j=0; j<n; ++j)
cin>>x.a[i][j];
t=powmatrix(x,k);
ans=0;
for(int i=0; i<n; ++i)
ans+=t.a[i][i],ans%=mod;
cout<<ans<<endl;
}
return 0;
}