HDU 1575 Tr A 矩阵快速幂

                                                Tr A

                          Time Limit: 1000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)


 

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的内容。

 

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

 

//矩阵快速幂

#include<cstdio>
#include<cstring>
using namespace std;
const int N=10;
int tmp[N][N];
int a[10][10];
void multi(int a[][N],int b[][N],int n) {   //  矩阵乘法
	memset(tmp,0,sizeof tmp);
	for(int i=0; i<n; i++)
		for(int j=0; j<n; j++)
			for(int k=0; k<n; k++) {
				tmp[i][j]+=a[i][k]*b[k][j];
				tmp[i][j]=tmp[i][j]%9973;
			}
	for(int i=0; i<n; i++)
		for(int j=0; j<n; j++)
			a[i][j]=tmp[i][j];
}
int res[N][N];
void Pow(int a[][N],int n) {
	memset(res,0,sizeof res);
	for(int i=0; i<N; i++) res[i][i]=1;
	while(n) {
		if(n&1)
			multi(res,a,N);
		multi(a,a,N);
		n>>=1;
	}
}
int main() {
	int t;
	scanf("%d",&t);
	while(t--) {
		int n,k;
		scanf("%d%d",&n,&k);
		for(int i=0; i<n; i++) {
			for(int j=0; j<n; j++) {
				scanf("%d",&a[i][j]);
			}
		}
		Pow(a,k);
		int sum=0;
		for(int i=0; i<n; i++) {
			sum+=res[i][i];
			sum=sum%9973;
		}
		printf("%d\n",sum);
	}
}