Tr A（快速幂）

题目描述:

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。

样例输入:

3
1 42
7 
2 6335
0 9 
4 8 
3 29359
2 4 5 
5 1 7 
1 1 5 

样例输出:

3969
5510
5473

代码如下：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll T,n=0,k=0,a[101][101]={};
ll s[101][101]={};
void ju(ll b[101][101],ll c[101][101]){
	ll d[101][101]={};
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			for(int ii=1;ii<=n;ii++){
				d[i][j]=d[i][j]+(b[i][ii]*c[ii][j])%9973;
				d[i][j]=d[i][j]%9973;
			}
		}
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			b[i][j]=d[i][j]%9973;
		}
	}
}
void kui(ll k){
	for(int i=1;i<=n;i++){
		s[i][i]=1;
	}
	while(k){
		if(k&1){
			ju(s,a);
		}
		ju(a,a);
		k=k/2;
	}
}
int main(){
	cin>>T;
	while(T--){ 
		cin>>n>>k;
		
		for(int i=1;i<=n;i++){
			for(int j=1;j<=n;j++){
				cin>>a[i][j];
			} 
		} 
		kui(k);
		int num=0;
		for(int i=1;i<=n;i++){
			num=num+s[i][i];
			num=num%9973;
		}
		cout<<num%9973<<endl;
		for(int i=1;i<=n;i++){
			for(int j=1;j<=n;j++){
				a[i][j]=0;
				s[i][j]=0;
			} 
		}
	}
	return 0;
}