A为一个方阵,则Tr A表示A的迹(就是主对角线上各项的和),现要求Tr(A^k)%9973。
每组数据的第一行有n(2 <= n <= 10)和k(2 <= k < 10^9)两个数据。接下来有n行,每行有n个数据,每个数据的范围是 0,9 0,9,表示方阵A的内容。
2 2 2 1 0 0 1 3 99999999 1 2 3 4 5 6 7 8 9
22686
矩阵的快速幂运算要注意和记住的以下几点:
1:矩阵相乘函数中这里最重要:temp[i][j]=(temp[i][j]+ak[i][k]*bk[k][j])%mod;
2:快速幂函数中单位矩阵的初始化
就以上两点值得注意吧。
#include<bits/stdc++.h> #define mod 9973 using namespace std; int T,n,k; int a[11][11]; void multi(int ak[][11],int bk[][11]) { int temp[11][11]; memset(temp,0,sizeof(temp)); int i,j,k; for(i=0;i<n;i++) for(j=0;j<n;j++) for(k=0;k<n;k++) { temp[i][j]=(temp[i][j]+ak[i][k]*bk[k][j])%mod; } for(i=0;i<n;i++) for(j=0;j<n;j++) ak[i][j]=temp[i][j]; } int quick(int ak[][11]) { int i,j; int E[11][11]; memset(E,0,sizeof(E)); for(i=0;i<11;i++) E[i][i]=1; while(k>0) { if(k&1) { multi(E,ak); } k>>=1; multi(ak,ak); } for(i=0;i<n;i++) for(j=0;j<n;j++) ak[i][j]=E[i][j]; } int main() { int i,j; cin>>T; while(T--) { cin>>n>>k; for(i=0;i<n;i++) for(j=0;j<n;j++) cin>>a[i][j]; quick(a); int ans=0; for(i=0;i<n;i++) ans=(ans+a[i][i])%mod; printf("%d\n",ans); } return 0; }