矩阵快速幂模板HDU——1575

#include <iostream>#include<vector>#include <cstdio>using namespace std;//用二维vector表示矩阵typedef vector<int> vec;typedef vector<vec> mat;typedef long long ll;const int mod=9973;//计算A*Bmat mul(mat &A,mat &B){mat C(A.size(),vec(B[0].size()));for(int i=0;i<A.size();i++){for(int k=0;k<B.size();k++){for(int j=0;j<B[0].size();j++){C[i][j]=(C[i][j]+A[i][k]*B[k][j])%mod;}}}return C;}//计算A^nmat pow(mat A,ll n){mat B(A.size(),vec(A.size()));for(int i=0;i<A.size();i++)B[i][i]=1;while(n>0){if(n&1)B=mul(B,A);A=mul(A,A);n>>=1;}return B;}int main(){int T;cin>>T;int n,k;while(T--){scanf("%d%d",&n,&k);mat A(n,vec(n));for(int i=0;i<n;i++)for(int j=0;j<n;j++){scanf("%d",&A[i][j]);}A=pow(A,k);int tr=0;for(int i=0;i<n;i++)tr=(tr+A[i][i])%mod;printf("%d\n",tr);}return 0;}