hdu 1575 Tr A

Tr A Time Limit: 1000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others) Total Submission(s): 5238    Accepted Submission(s): 3927 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<iostream> #include<cmath> #include<algorithm> #include<cstring> #include<iomanip> #include<cstdio> using namespace std; long long int sq[11][11]; long long int sq1[11][11]; long long int sq2[11][11]; int main(){ int T; cin>>T; int v,mi; while(T--) { cin>>v>>mi; int i,j; for(i=1;i<=v;i++) for(j=1;j<=v;j++) { cin>>sq[i][j]; sq2[i][j]=sq[i][j]; } mi--; while(mi>=1) { if(mi%2==1) { mi--; for(i=1;i<=v;i++) { for(j=1;j<=v;j++) { long long int sum=0; for(int s=1;s<=v;s++) sum=(sum+sq[i][s]*sq2[s][j]%9973)%9973; sum=sum%9973; sq1[i][j]=sum; } } for(i=1;i<=v;i++) for(j=1;j<=v;j++) sq2[i][j]=sq1[i][j]; } if(mi%2==0) { mi=mi/2; for(i=1;i<=v;i++) { for(j=1;j<=v;j++) { long long int sum=0; for(int s=1;s<=v;s++) sum=(sum+sq[i][s]*sq[s][j]%9973)%9973; sum=sum%9973; sq1[i][j]=sum; } } for(i=1;i<=v;i++) for(j=1;j<=v;j++) sq[i][j]=sq1[i][j]; } } long long int ans=0; for(i=1;i<=v;i++) { ans=(ans+sq2[i][i])%9973; } cout<<ans<<endl; } return 0; }