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 <bits/stdc++.h>
using namespace std;
int mod = 9973;
int len,k;
struct P
{
int mat[15][15];
};
P x;
P cal(P a,P b)//将a和b乘起来
{
P c;
for(int i=0; i<len; i++)
for(int j=0; j<len; j++)
{
c.mat[i][j]=0;
for(int k=0;k<len;k++)
c.mat[i][j]=(c.mat[i][j]+a.mat[i][k]*b.mat[k][j]%mod)%mod;
}
return c;
}
P calculate(P a,P b,int c)//计算b*a^c
{
while(c)
{
if(c&1)
b=cal(b,a);
c>>=1;
a=cal(a,a);
}
return b;
}
int main()
{
// freopen("in.txt","r",stdin);
int T;
scanf("%d",&T);
while(T--)
{
scanf("%d%d",&len,&k);
for(int i=0;i<len;i++)
for(int j=0;j<len;j++)
scanf("%d",&x.mat[i][j]);
int ans=0;
x=calculate(x,x,k-1);//减一之后才是正确结果x*x^(k-1);
for(int i=0;i<len;i++)
ans=(ans+x.mat[i][i])%mod;
printf("%d\n",ans);
}
return 0;
}
【续】题目不难,但是我开始不是这样写的,看了几位大佬之后才这样写的,主要是这个结构体我没想到,确实,因为结构体可以直接用=的操作确实很方便,而不是循环。