题目链接 http://acm.hdu.edu.cn/showproblem.php?pid=1575
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 <algorithm>
#include <cstring>
using namespace std;
const int mod = 9973;
int n,k;
// 定义方阵
struct Matrix {
int data[15][15];
Matrix() {memset(data,0,sizeof(data));}
friend Matrix operator * (const Matrix &a,const Matrix &b) {
Matrix t;
for(int i = 1;i <= n;i++)
for(int j = 1;j <= n;j++)
for(int m = 1;m <= n;m++) {
t.data[i][j] = (t.data[i][j] + a.data[i][m]*b.data[m][j]) % mod;
}
return t;
}
}mat;
Matrix Multi(Matrix mat,int k) {
if(k == 1)
return mat;
int mid = k / 2;
Matrix ret = Multi(mat,mid);
if(k & 1) {
// k是奇数
return ret * ret * mat;
} else {
return ret * ret;
}
}
int main()
{
int t;
scanf("%d",&t);
while(t--) {
scanf("%d%d",&n,&k);
for(int i = 1;i <= n;i++)
for(int j = 1;j <= n;j++) {
scanf("%d",&mat.data[i][j]);
}
Matrix ans = Multi(mat,k);
int result = 0;
for(int i = 1;i <= n;i++)
result += ans.data[i][i];
printf("%d\n",result%mod);
}
return 0;
}